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

Yolov4-tiny-3l.cfg #353

Closed
Haerin0 opened this issue May 21, 2021 · 4 comments
Closed

Yolov4-tiny-3l.cfg #353

Haerin0 opened this issue May 21, 2021 · 4 comments

Comments

@Haerin0
Copy link

Haerin0 commented May 21, 2021

I trained my custom dataset with yolov4-custom.cfg, yolov4-tiny-custom.cfg and yolov4-tiny-3l.cfg.
it worked fine except yolov4-tiny-3l.cfg. it doesn't shows any bounding boxes at all. is that mean it can't detect anything? if so, which part of yolov4-tiny-3l.cfg should i edit? Thank you.

@aditya-5842
Copy link

I think yolov4-tiny-3l is not yet implemented in the repo. I have made the necessary for this one and it's is working for me.

Have a look at this: https://github.com/aditya-5842/tensorflow-yolov4-tflite OR you can look into #325

@kaijaz123
Copy link

kaijaz123 commented Jul 17, 2021

@aditya-5842 Hi good day. I am using your repo https://github.com/aditya-5842/tensorflow-yolov4-tflite to convert the yolov4-tiny-3l weights. However, it detects nothing when I run the detectvideo. Did I need to change anything?

Command to convert:
!python save_model.py --weights weights/yolov4-tiny-3l_best.weights --input_size 416 --tiny --model yolov4 --output checkpoints/3l_ball_416

@aditya-5842
Copy link

@kaijaz123 There is a small change in the command that you are using.

python save_model.py --weights ./data/yolov4-tiny-3l-608_5000.weights --output ./checkpoints/yolov4-tiny-3l-608 --input_size 608 --model yolov4-tiny-3l --tiny

Thank you for using my repo. If you will notice there is another branch in my repo, there I have these made change. You can find the above command and necessary changes in the yolov4-tiny-3l-fix branch.

If you're trained model predicts multiple classes, please make necessary changes here:

https://github.com/aditya-5842/tensorflow-yolov4-tflite/blob/yolov4-tiny-3l-fix/core/config.py#L14

@kaijaz123
Copy link

kaijaz123 commented Jul 17, 2021

@aditya-5842 Hi, sorry for that. I didnt notice there was another branch as I was using the original code. It is working and able to detect now. Thanks man!

@Haerin0 Haerin0 closed this as completed Dec 16, 2021
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

3 participants