预测会用到两个文件,分别是yolo.py和predict.py。
yolo.py里面修改model_path以及classes_path,这两个参数必须要修改。
model_path指向训练好的权值文件,在logs文件夹里。
classes_path指向自己目标类别所对应的txt。
predict.py也要修改图片路径和保存路径。该代码可以选择不同模式进行单张图片、多张图片、视频及FPS的预测。
修改好后,运行predict.py即可完成检测。
预测会用到两个文件,分别是yolo.py和predict.py。
yolo.py里面修改model_path以及classes_path,这两个参数必须要修改。
model_path指向训练好的权值文件,在logs文件夹里。
classes_path指向自己目标类别所对应的txt。
predict.py也要修改图片路径和保存路径。该代码可以选择不同模式进行单张图片、多张图片、视频及FPS的预测。
修改好后,运行predict.py即可完成检测。