search

SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

YOLO ModelCompression MultidatasetTraining
GNU General Public License v3.0
444 stars 136 forks source link

YOLOv4 #101

Open CaiyuZhang opened 3 years ago

CaiyuZhang commented 3 years ago

你好，请问如果我要先用yolov4训练自己的数据集，然后再使用您的repo来进行模型剪枝，那么这整体的步骤是什么呢，后续能否出个yolov4的教程什么的。是不是先用AlexAB的darknet训练自己的yolov4模型，然后将自己数据集的cfg和训练好的weights权重在您的repo去做稀疏化训练和剪枝，谢谢了。

SpursLipu commented 3 years ago

不用的 yolov4的步骤完全和yolov3一样的，你自己写好cfg文件，然后用我提供的yolov4预训练文件然后加指令-pt训练就可以了