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

Transfer Learning on YOLOv4-CSP #386

Open
srikhetramohanty opened this issue May 3, 2022 · 0 comments
Open

Transfer Learning on YOLOv4-CSP #386

srikhetramohanty opened this issue May 3, 2022 · 0 comments

Comments

@srikhetramohanty
Copy link

srikhetramohanty commented May 3, 2022

What layers need to be freezed to perform transfer learning on YOLOv4-csp effectively? Currently I have frozen all the layers except the 3 conv layers before YOLO layers. ~31k trainable parameters on my custom dataset with ~400 training images and 1 class. The loss doesnt seem to reduce and the precision recall remains at 0 till ~200-300 epochs. But when i train from scratch on the same data, it achieves decent performance of ~0.65 mAP(0.5). Any way out? @WongKinYiu Thanks in advance.

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

1 participant