Function optimize

scripts/optimize_bboxes.py:124–154

(full_bboxes, optim_idx, batch_size=5000)

def optimize(full_bboxes, optim_idx, batch_size=5000):
    global batch_idx, scales, aspect_ratios, conv_sizes

    start = batch_idx * batch_size
    end   = min((batch_idx + 1) * batch_size, full_bboxes.size(0))

    if batch_idx > (full_bboxes.size(0) // batch_size):
        batch_idx = 0

    bboxes = full_bboxes[start:end, :]

    anchor_base = [
        make_priors(conv_sizes[idx], scales[idx], aspect_ratios[idx])
            for idx in range(len(conv_sizes)) if idx != optim_idx]
    base_hits = compute_hits(bboxes, torch.cat(anchor_base, dim=0))
    
    
    def set_x(x, scales, aspect_ratios):
        if optimize_scales:
            for i in range(len(scales)):
                scales[i] = max(x[i], 0)
        else:
            k = 0
            for i in range(len(aspect_ratios)):
                for j in range(len(aspect_ratios[i])):
                    aspect_ratios[i][j] = x[k]
                    k += 1
            

    res = minimize(step, x0=scales[optim_idx] if optimize_scales else sum(aspect_ratios[optim_idx], []), method='Powell',
        args = (set_x, bboxes, base_hits, optim_idx),)


def pretty_str(x:list):