yjh0410 / RT-ODLab

YOLO Tutorial
MIT License
136 stars 29 forks source link

nothing output when I run demo.py using yolov1/yolov2/yolov3 pretrained model #14

Open Rainy000 opened 6 months ago

Rainy000 commented 6 months ago

Hi,thank you for your great work! I try to use your code and test image to show the detection result, but nothing be detected.

my environment below:

platform: windows 11 python version: 3.6.13 torch: 1.9.1 pretrained model: download from the url in readme

yjh0410 commented 6 months ago

@Rainy000 Thanks for your approval. You might add --show in your command running demo.py, otherwise you won't see the detection results.

Rainy000 commented 6 months ago

@Rainy000 Thanks for your approval. You might add --show in your command running demo.py, otherwise you won't see the detection results.

vis detection

img_processed = visualize(image=image, 
                          bboxes=bboxes,
                          scores=scores, 
                          labels=labels,
                          class_colors=class_colors,
                          class_names=class_names,
                          class_indexs=class_indexs)
cv2.imwrite(os.path.join(save_path, str(i).zfill(6)+'.jpg'), img_processed)

I mean the saved image of img_processed has nothing (not exist any boxes)

yjh0410 commented 6 months ago

@Rainy000 Thanks for your feedback. We have fixed this bug. You can try again.

Rainy000 commented 6 months ago

@yjh0410 It‘s ok, thanks