theAIGuysCode / YOLOv4-Cloud-Tutorial

This repository walks you through how to Build and Run YOLOv4 Object Detections with Darknet in the Cloud with Google Colab.
355 stars 300 forks source link

Remove output layer of trained Yolo #11

Open Mshz2 opened 4 years ago

Mshz2 commented 4 years ago

Hey there,

Thanks for the tutorial. I would like to first train the Yolo with large dataset relevant to my target class and then do transfer learning with with small dataset as the target class.

In the description, the yolov4.conv.137 which trained with coco dataset is given for doing transfer learning. Would U please help me, how can I remove the output layer of initial trained yolo of my own and get only yolo.conv weights like the given one?

Best