PyTorch_YOLOv4 icon indicating copy to clipboard operation
PyTorch_YOLOv4 copied to clipboard

backbone pretrained model 詢問

Open peggy860629 opened this issue 4 years ago • 14 comments

@WongKinYiu 您好:

想請教有提供backbone pretrained model嗎? 例如我想訓練自己的detector(yolo layer), 但是backbone(cspdarknet53)的部分不想重新訓練. 我對object detection models的理解是會先用backbone在ImageNet 上training完成, 再與detector 整合end to end 在MSCOCO上做training

謝謝

peggy860629 avatar Nov 05 '20 07:11 peggy860629

這裡所有結果都是train from scratch, 沒有使用imagenet pretrain

WongKinYiu avatar Nov 06 '20 13:11 WongKinYiu

有没有增加backbone在ImageNet上预训练权重的,最后的检测结果mAP和train from scratch的结果相比能提高多少?

sunshine-zkf avatar Nov 11 '20 02:11 sunshine-zkf

@WongKinYiu 您好:

想請教有提供backbone pretrained model嗎? 例如我想訓練自己的detector(yolo layer), 但是backbone(cspdarknet53)的部分不想重新訓練. 我對object detection models的理解是會先用backbone在ImageNet 上training完成, 再與detector 整合end to end 在MSCOCO上做training

謝謝

您好,请问您现在找到backbone部分的pretrain model了吗,我也希望能利用backbone的imagenet model自己train一遍

xuzhang1199 avatar Jun 16 '21 02:06 xuzhang1199

最近剛完成imagenet訓練, 還沒train在coco上.

WongKinYiu avatar Jun 16 '21 04:06 WongKinYiu

最近剛完成imagenet訓練, 還沒train在coco上.

请问cspdarknet53在imagenet上面的weight能发吗

xuzhang1199 avatar Jun 16 '21 06:06 xuzhang1199

LeakyReLU版的weights這裡有 https://github.com/WongKinYiu/CrossStagePartialNetworks#big-models

WongKinYiu avatar Jun 16 '21 06:06 WongKinYiu

LeakyReLU版的weights這裡有 https://github.com/WongKinYiu/CrossStagePartialNetworks#big-models

谢谢您

xuzhang1199 avatar Jun 16 '21 06:06 xuzhang1199

LeakyReLU版的weights這裡有 https://github.com/WongKinYiu/CrossStagePartialNetworks#big-models

有mish版本的吗

xuzhang1199 avatar Jun 16 '21 06:06 xuzhang1199

https://github.com/WongKinYiu/CrossStagePartialNetworks/blob/master/imagenet/results.md#cspdarknet-53

WongKinYiu avatar Jun 16 '21 06:06 WongKinYiu

https://github.com/WongKinYiu/CrossStagePartialNetworks/blob/master/imagenet/results.md#cspdarknet-53

非常感谢!

xuzhang1199 avatar Jun 16 '21 06:06 xuzhang1199

要注意一點, cspdarknet53 是 yolov4 的 backbone, 和 yolov4-csp 的 backbone 不太一樣.

WongKinYiu avatar Jun 16 '21 06:06 WongKinYiu

要注意一點, cspdarknet53 是 yolov4 的 backbone, 和 yolov4-csp 的 backbone 不太一樣. 谢谢

xuzhang1199 avatar Jun 16 '21 07:06 xuzhang1199

imagenet pretrain沒有比較好

WongKinYiu avatar Jun 21 '21 06:06 WongKinYiu

我想问一下,训练coco数据集花了多长时间?

q1135718080 avatar Mar 21 '22 02:03 q1135718080