chenyuntc / simple-faster-rcnn-pytorch

A simplified implemention of Faster R-CNN that replicate performance from origin paper
Other
3.97k stars 1.14k forks source link

Strange images from the nwtwork #184

Open ZichengDuan opened 4 years ago

ZichengDuan commented 4 years ago

Hi to everyone, I encounter a strange issue. Due to some reason, I am not able to use visdom on my PC so I was trying to save the images and bboxes by hands directly onto the image, however, I got something like this: faster The shape for this image is (1, 3, 600, 800), and I printed it bu using cv2.imwrite(). I tracked the source of the image and it seems that it comes from class Dataloader. The predict function in faster_rcnn.py is also using this image for prediction. I am very confused and I need your help, thank you very much!