Skip to content

v1.0
Compare
Choose a tag to compare
@yyccR yyccR released this 21 Jun 09:29
· 12 commits to master since this release
v1.0
e86641c

v1.0 几个总结：

  • [1]. 模型结构总的与 ultralytics/yolov5 v6.0 保持一致
  • [2]. 其中Conv层替换swish为Relu
  • [3]. 整体数据增强与 ultralytics/yolov5 保持一致
  • [4]. readme中训练所需的数据集为kaggle公开猫狗脸检测数据集，已放到release列表中
  • [5]. 为什么不训练coco数据集？因为没资源，跑一个coco要很久的，服务器一直都有任务在跑所以没空去跑 - . -
  • [6]. release里的yolov5s-best.h5为上述kaggle猫狗脸数据集的训练权重，训练:测试为8:2，val精度大概如下：
class mAP@0.5 mAP@0.5:0.95 precision recall
cat 0.905156 0.584378 0.682848 0.886555
dog 0.940633 0.513005 0.724036 0.934866
total 0.922895 0.548692 0.703442 0.910710

就这样，继续加油！💪🏻💪🏻💪🏻

Assets 4
Loading