YOLOX 学习
本文仅将yolox跑起来,不涉及到原理性知识。
先看成果
使用VOC格式的数据集进行训练:
使用COCO格式数据得出的结果:
学习路线
踩过的坑&Solution&心得
- 遇到bug,应该先从Issues里面去寻找解决办法,像这种很多人用的项目,踩过的坑99%都能在这里寻找到Solution.
- Linux跑完模型之后,需要将进程kill掉,使用下面这两行命令即可
fuser -v /dev/nvidia* // 查看运行的进程
kill -9 -1 // kill掉运行的进程
- CUDA版本 - CUDANN版本 - Pytorch版本是有对应关系的
- ModuleNotFoundError: No module named ‘yolox‘
这个bug本次实验出现了很多次
Solution1:
Solution2:
pip install -v -e .
- COCO的目录格式需要
- “Defaulting to user installation because normal site-packages is not writeable”
Solution:https://zhuanlan.zhihu.com/p/411025149