Skip to content

How to infer one Image with COCO Config? #144

@radarjonathan

Description

@radarjonathan

Hey Guys,

i'm currently trying to use the networt with coco_4_gpu on infering one image.

I'm inputting:
-jpg image with 1280x720
img = cv2.imread('00000001.jpg')
img = cv2.resize(img,(1280,720),interpolation=cv2.INTER_CUBIC)
img = img.transpose(2, 0, 1)
img_tensor = torch.from_numpy(img)
img_tensor = torch.unsqueeze(img_tensor,0).type(torch.FloatTensor).cuda()
img_info = np.array([[720, 1280, 3]])
data = {'data': img_tensor , 'img_info' : img_info}

When im running the model output i get:

88 fpn_p5_upsample = self.fpn_upsample(fpn_p5_1x1)
---> 89 fpn_p4_plus = fpn_p5_upsample + fpn_p4_1x1
90 fpn_p4_upsample = self.fpn_upsample(fpn_p4_plus)
91 fpn_p3_plus = fpn_p4_upsample + fpn_p3_1x1

RuntimeError: The size of tensor a (46) must match the size of tensor b (45) at non-singleton dimension 2

Whats the matter?

Im on Pytorch 1.4.

Kindest regards!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions