Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
Rainy000 opened this issue Mar 3, 2024 · 4 comments

Comments

@Rainy000
Copy link

Rainy000 commented Mar 3, 2024

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
Copy link
Owner

yjh0410 commented Mar 3, 2024

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

@Rainy000
Copy link
Author

Rainy000 commented Mar 3, 2024

@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
Copy link
Owner

yjh0410 commented Mar 3, 2024

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

@Rainy000
Copy link
Author

Rainy000 commented Mar 4, 2024

@yjh0410 It‘s ok, thanks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants