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

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

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

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

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

你可能感兴趣的:(神经网络,深度学习,python)