神经网络学习----如何利用训练好的神经网络模型进行预测

预测会用到两个文件,分别是yolo.py和predict.py。

yolo.py里面修改model_path以及classes_path,这两个参数必须要修改。
model_path指向训练好的权值文件,在logs文件夹里。
classes_path指向自己目标类别所对应的txt。

predict.py也要修改图片路径和保存路径。该代码可以选择不同模式进行单张图片、多张图片、视频及FPS的预测。

修改好后,运行predict.py即可完成检测。


版权声明:本文为qq_45630900原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。