pytorch_solov2
pytorch_solov2 copied to clipboard
作者能不能说一说代码还有什么缺陷,方便我们研究补全,谢谢了
缺陷还挺多的,现在工作比较忙,没时间维护了T_T
- 只支持resnet18、resnet34两种backbone的训练和测试.
- 不支持多显卡并行训练
- 训练和测试的配置项不完整
- eval时没有可视化
- 导出ONNX的脚本,后续部署流程
shortages:
- Only supports Resnet18, Resnet34 backbone to train.
- Multi GPU parallel training is not supported.
- Incomplete configuration items for training and testing.
- visualization
- export onnx model
@seekerwmy 最好参照官方代码做一个对比:smile:
谢谢作者