Index Error when using Transfer Learning about objectdetection HOT 1 OPEN

detect_thresh = 0.3
nms_thresh = 0.5
size = 1024

nr = []
pos = []
for image_id, (test_image, emp_label) in enumerate(learn.data.test_ds):
    #mean, std = learn.data.stats

    pos.append([0, 0, 0, 0])
    nr.append(0)

    image_ori = test_image
    # ori_shape = image_ori.shape[:2]

    # image = cv2.resize(image_ori, (size, size))

    # image = pil2tensor(image / 255., np.float32).cuda()

    #image = transforms.Normalize(mean, std)(image)
    
    image = test_image.resize(1024).data.cuda()

    # class_pred_batch, bbox_pred_batch, _ = learn.model(image[None, :, :, :])
    
    class_pred_batch, bbox_pred_batch, _ = learn.model(image[None, :])

    for clas_pred, bbox_pred in zip(class_pred_batch, bbox_pred_batch):
        bbox_pred, scores, preds = process_output(clas_pred, bbox_pred, anchors, detect_thresh)

        if bbox_pred is not None:
            to_keep = nms(bbox_pred, scores, nms_thresh)
            bbox_pred, preds, scores = bbox_pred[to_keep].cpu(), preds[to_keep].cpu(), scores[to_keep].cpu()

            t_sz = torch.Tensor(ori_shape)[None].float()
            bbox_pred = rescale_box(bbox_pred, t_sz)

            temp_pos = [0, 0, 0, 0]
            for id, box in enumerate(bbox_pred[:2]):
                x = to_np(box[0] + box[2] / 2)
                y = to_np(box[1] + box[3] / 2)
                if id == 0:
                    temp_pos[ :2] = x,y
                    nr[image_id] = 1
                else:
                    temp_pos[2: ] = x, y
                    nr[image_id] = 2

RuntimeError                              Traceback (most recent call last)
<ipython-input-65-883a98e5b6df> in <module>
     27 
     28     for clas_pred, bbox_pred in zip(class_pred_batch, bbox_pred_batch):
---> 29         bbox_pred, scores, preds = process_output(clas_pred, bbox_pred, anchors, detect_thresh)
     30 
     31         if bbox_pred is not None:

D:\ProgramFile\anaconda\envs\ai\lib\site-packages\object_detection_fastai\helper\object_detection_helper.py in process_output(clas_pred, bbox_pred, anchors, detect_thresh)
    227 
    228 def process_output(clas_pred, bbox_pred, anchors, detect_thresh=0.25):
--> 229     bbox_pred = activ_to_bbox(bbox_pred, anchors.to(clas_pred.device))
    230     clas_pred = torch.sigmoid(clas_pred)
    231 

D:\ProgramFile\anaconda\envs\ai\lib\site-packages\object_detection_fastai\helper\object_detection_helper.py in activ_to_bbox(acts, anchors, flatten)
     74     if flatten:
     75         acts.mul_(acts.new_tensor([[0.1, 0.1, 0.2, 0.2]]))
---> 76         centers = anchors[...,2:] * acts[...,:2] + anchors[...,:2]
     77         sizes = anchors[...,2:] * torch.exp(acts[...,2:])
     78         return torch.cat([centers, sizes], -1)

RuntimeError: The size of tensor a (24480) must match the size of tensor b (24192) at non-singleton dimension 0