PaddleClas预测结果默认保存在哪里?
收藏
python3 tools/infer.py \
-c ./ppcls/configs/quick_start/MobileNetV3_large_x1_0.yaml \
-o Infer.infer_imgs=dataset/flowers102/jpg/image_00001.jpg \ #待预测的图片文件路径或者批量预测时的图片文件夹
-o Global.pretrained_model=./output/MobileNetV3_large_x1_0/best_model
PaddleClas训练好后,运行以上代码,可以看到打印出的预测结果。
问一下预测结果默认保存在哪个变量或文件里了?如果没有保存的话怎样指定保存位置?
jsdbzcm
已解决
2#
回复于2022-01
自己搞定了,修改PaddleClas/ppcls/engine/engine.py文件,在infer函数中添加3行(位置见备注),就可以把预测结果保存在result.npy文件中了。 [代码]
1
收藏
请登录后评论
自己搞定了,修改PaddleClas/ppcls/engine/engine.py文件,在infer函数中添加3行(位置见备注),就可以把预测结果保存在result.npy文件中了。
应该是可以指定参数,直接保存出一个result.txt的,可以看看对应的参数。
学习了,谢谢大佬们
学习了~
paddleclas2.1版本的文件组织结构好像不一样,这样改的:
改文件PaddleClas/tools/infer/infer.py
然后执行infer.py脚本,最后在/home/aistudio/目录下生成result_test.csv文件
学习了