-
Notifications
You must be signed in to change notification settings - Fork 6
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
use other yolov5 models #5
Comments
Hi, actually I have not tried to apply DRENet with larger backbone than YOLOv5s previously. But if you want to, it is very easy. Just change the depth_multiple and width_multiple in the .yaml to apply which size model you want, these two params just corresponds to the number of layers and the channels of each module. You can refer to the yamls in YOLOv5 official repo and the usage of these two params in the code for more details. Furthermore, for large model, you may need to try different number of C3ResAtnMHSA layers in the yaml for a better result. Current for YOLOv5s is 1 = 3 * 0.33. |
For applying different size of YOLOv5, such as YOLOv5m, YOLOv5l, etc., you only need to change these two params in DRENet.yaml: E.g., YOLOv5s is 0.33 and 0.5 respectively, YOLOv5m is 0.67 and 0.75, etc. You can refer to more details in Yolov5 official Repo. Then, you have applied DRENet with different size of YOLOv5. About the number of C3ResAtnMHSA, for the current configuration, it is |
Hi @WindVChen thank you so much for your help. I think your method works better in my dataset as compared to original yolov5 implementation. Can I use this repo for further implementation ? or may be to use it in data science competition? thank you |
I'm glad that the method can help. Just feel free to use it as long as the source is indicated. |
Hi, I wonder if I can use other yolov5 model (n,s,m,l,etc)? I have tried to use DRENet.yaml inside models folder. is it possible to use yaml file from yolov5 repo? do I need to do some modification?
please advise
thanks
The text was updated successfully, but these errors were encountered: