通过flow_from_directory读取了三个类别的图片数据,训练了一个图像分类模型,请问如何用这个模型预测未知标签的图片——如何获得满足输入层要求的图片数据?
1条回答 默认 最新
爱晚乏客游 2022-11-11 13:33关注不是,你都能训练了,为啥调用模型识别不会呢?这不就是你在train里面计算loss的时候,model(input)的返回结果吗?你的input怎么来的,预测的时候一样处理图片就行,只不过batch size可以设置为1就是了。
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报