数据集下载地址百度网盘(提取码:ippg),下载后解压到wrist_images
文件夹下。
在YOLO下载一个weights文件,放在跟目录下,命名为yolov3.weights。
将weights文件转化为模型文件:
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
运行
python train.py
进行训练。
训练完成后,运行
python yolo_inference_miou.py --image
进行模型评价。测试结果:平均IOU为0.66,最大IOU为0.95。数据集只有较少的图片,扩大数据集的数量可得到更好的结果。