使用提供的数据集重新训练squeezenet1_1,效果十分差
- 下载并解压了README中提供的数据集“该项目用到的制作数据集下载地址(百度网盘 Password: ara8 )”
- 在train.py中定位数据集路径并进行训练:python train.py
- 3000个epoch后,best_loss为0.075815
预览结果如
其他图片的情况也是一样的,全部关键点在图片的左上角的固定位置。
请问是哪里遗漏了步骤吗?
实验环境:
Ubuntu20.04 + python3.7 + pytorch1.5.1 + RTX2060