search

ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"
https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/
2.32k stars 725 forks source link

'NoneType' object has no attribute 'shape' #62

Open anjanaouseph opened 4 years ago

anjanaouseph commented 4 years ago

Hey first of all thanks for this great tutorial! I have encountered this error while running the code:

PyTorch Variables for images

im_batches = list(map(prep_image, loaded_ims, [inp_dim for x in range(len(imlist))]))

List containing dimensions of original images

im_dim_list = [(x.shape[1], x.shape[0]) for x in loaded_ims] im_dim_list = torch.FloatTensor(im_dim_list).repeat(1,2)

AttributeError Traceback (most recent call last)

in () ----> 1 im_batches = list(map(prep_image, loaded_ims, [inp_dim for x in range(len(imlist))])) 2 3 #List containing dimensions of original images 4 im_dim_list = [(x.shape[1], x.shape[0]) for x in loaded_ims] 5 im_dim_list = torch.FloatTensor(im_dim_list).repeat(1,2) 1 frames in letterbox_image(img, inp_dim) 1 def letterbox_image(img, inp_dim): 2 '''resize image with unchanged aspect ratio using padding''' ----> 3 img_w, img_h = img.shape[1], img.shape[0] 4 w, h = inp_dim 5 new_w = int(img_w * min(w/img_w, h/img_h)) AttributeError: 'NoneType' object has no attribute 'shape' --------------------------------------------------------------------------------------------------------------How can this be resolved?
kalikhademi commented 1 year ago

You should double check if the loaded image has properly read the opencv image. I had the same error and when I checked it did not have the absolute path to read the image to make imlist and later loaded_ims.