台球测试2:
学习内容:
对台球数据集进行训练和测试,代码均来自yunyang的github。
一、Tensorflow2.0版本的代码
继续研究该代码,尝试使用不同数量的数据集进行训练,结果依旧未测试出测试集中的图片的目标
查资料后认为:
$ git clone https://github.com/YunYang1994/yymnist.git
$ python yymnist/make_data.py --images_num 1000 --images_path ./data/dataset/train --labels_txt ./data/dataset/yymnist_train.txt
$ python yymnist/make_data.py --images_num 200 --images_path ./data/dataset/test --labels_txt ./data/dataset/yymnist_test.txt
该处可能是问题的原因(上图),暂时先不研究该代码(之后再看)
寻找新代码
二、Tensorflow1.x版本的代码
1.问题一
将台球的数据集进行训练,未测试出结果
解决:
查找资料,网上有博主使用该代码对其自己的数据集进行训练,且可以测试出结果,但是,也有一些人出现与我类似的问题
评论提问博主,博主认为是没有识别出来
2.总结一
训练台球数据集最后的loss值是11.9x,较大
思考认为,是数据集数量的问题,寻找较大的数据集(VOC数据集)进行训练
训练中。。。
3.问题二
可能是数据集准备有问题(需要修改,未改)
看了几篇博客,越来越觉得是数据集在准备的时候出现了问题,等上面的数据集训练完成后进行尝试
停止训练。。。
数据集太大,训练了8个epoch,大概用了20多个小时
使用目前训练出来的权重进行测试,有效果
接下来解决问题二
从网上找到一个红细胞的数据集,通过该模型进行训练,训练之后进行测试,失败,未找到目标。。。。。。。。。。。
利用训练出来的权重测试单张图片,可以测试成功,测试测试集则无法预测
4.问题三
单张照片可以预测出结果,说明权重是对的,问题出现在测试test集的代码上
解决:
查看并修改对应的代码,问题解决,可以找到红细胞目标
5.问题四
使用台球训练集进行训练并测试
测试结束
test集包括3张train集中的图片
图片如下:
train集图片进行测试:
压缩后的图片进行测试:
test集其他图片进行测试:
解决:
需增大数据集
6.总结二
由此推断出,之前的代码应该也是测试test集代码的问题,不过,之前的代码只能训练416x416大小的图片
台球集测试效果不佳,需增大train集进行训练