徐静

这个issue出现的可能性很多，仅上述文字描述无法给出你比较满意的解决办法，首先确定训练和测试代码是否有问题，如果没有问题，请确定训练过程是否正常，或训练数据构建是否正确。

请参考： ``` $ python3 detect.py --source file.jpg # image file.mp4 # video ./dir # directory 0 # webcam rtsp://170.93.143.139/rtplive/470011e600ef003a004ee33696235daa # rtsp stream http://112.50.243.8/PLTV/88888888/224/3221225900/1.m3u8 # http stream ``` 内有视频推断和保存完整的代码！！！

这里 有我们测试视频和保存视频的样例

已通过邮箱发送，请注意查收

请检查你的pytorch可以正常调用GPU，怀疑你的pytorch编译的CUDA版本和你机器装的CUDA版本不一致，导致pytorch不能使用GPU

``` import torch print(torch.cuda.is_available()) ``` 应该是返回:True 的请自查

请指定使用的device

训练图片也只是用了1张？可否详细的说一下你训练集的情况？

我们有尝试过使用1张图像训练模型，个人观点： 训练集仅1张，模型无法再有限的训练集上学习更多的信息，同时我们也无法测试模型是否过拟合，这时的模型是学习不到好的参数的

一张图图像训练出来的网络，大概率上也预测不出来这张图像上的object, 因为你的网络没有做判断的能力，不用纠结了，一张图就别在训练网络了，没有意义，这些模型是建立在大量数据的基础上的。没有任何意义