1、在模型训练结束后,如下图所示,找到该文件夹。
2、然后找到weights文件夹中的best.pt文件,这就是该数据训练后的模型。
1、在assets文件夹下创建FPC-2文件夹,放入一些同类FPC预测结果。
2、和训练同级文件夹,找到predict.py文件,即为模型预测文件。
3、修改model路径,修改source文件夹名,改为我们刚才生成pt文件的文件夹和FPC-2图片文件夹。
4、运行predict.py文件,等待一会,直到显示白色字体的文件夹为止。
5、打开文件夹,即可查看预测结果