PyTorch YOLOv3训练自己的数据集–之–排坑指南

相比于官方给出的基于darknet框架的YOLOv3,基于PyTorch的YOLOv3源码易懂,使用更方便。我将自己在学习和探索过程中遇到的问题和解决的经验在这里与大家分享,希望能为初学者提供一些帮助。 1. 运行detect.py 时,出现RuntimeError:Invalid DISPLAY variable 解决方法:给detect.py的 import matplotlib 后面line22加 plt.switch_backend(‘agg’) 2. 训练时出现 UserWarning: indexing with dtype torch.uint8 is now deprecat