这周用yolov4干了两件事情,一个是训练tt100k交通标志数据集并测试,一个是用tt100k的图片标注出井盖,然后训练测试。
1. 训练tt100k交通标志数据集并测试
所用的机器:64G RAM; i7-10代;Nvidia GeForce 2080 super 8G
环境:CUDA 10.2 ; cudnn-8.0.2 ; python 3.7; openCV 3.4 ; yolov4
训练了两天左右,maxbatch设置为90000,训练到45000左右的时候意外断电,最高map为74%。
下图为博主用Nvidia Titan 11G 训练完成之后的结果
下面图片是我用训练完(一半)的模型测试的结果