search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

请问这个可以用VOC数据集训练吗 #166

Open Cai-ma opened 2 years ago

Cai-ma commented 2 years ago

我看指南上用的是coco数据集，如果我想用VOC，我该这么做？谢谢

AiXinYa commented 2 years ago

有一个voc_2007_for_yolo_torch的github仓库，里面有一套将voc格式转换为coco格式或者其他darknet能使用的代码。

chumingqian commented 2 years ago

Hi, @Cai-ma : How about try this https://github.com/chumingqian/Make_Dataset-for-Ultralytics-yolov3v4 .

Cai-ma commented 2 years ago

有一个voc_2007_for_yolo_torch的github仓库，里面有一套将voc格式转换为coco格式或者其他darknet能使用的代码。

谢谢，您的建议很有效

Cai-ma commented 2 years ago

有一个voc_2007_for_yolo_torch的github仓库，里面有一套将voc格式转换为coco格式或者其他darknet能使用的代码。

您好，我转成coco进行稀疏化训练的时候报错，说我缺少.shape文件，请问我该怎么得到这个文件？

AiXinYa commented 2 years ago

具体我忘记了，但是不需要用到这个.shape代码需要配合稀疏训练进行修改的地方比较多。根据报错来找就行