Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
994 stars 200 forks source link

预训练权重 #66

Open ruoyuchen0903 opened 2 years ago

ruoyuchen0903 commented 2 years ago

感谢大佬分享!问问大家和大佬,请问修改了backbone后是不是不宜再使用原来yolo5官方的预训练权重,我试了一下,大概掉了五个点。请问后续大佬会有在大数据集上训练后的权重分享嘛?如果不想自己从头开始训练,有什么方法嘛?