我已经使用 Flask 和 Heroku 部署了一个 CNN 模型。在代码模型的一部分中,模型需要一个图像,该图像将保存在一个文件夹名称上传中,然后图像的路径被提供给模型进行预测。当我在本地机器上运行模型时,它可以正常工作,但在 heroku 上显示 500 内部错误,可能是因为上传文件问题。这是我上传功能代码的一瞥。
def upload():
if request.method == 'POST':
# Get the file from post request
f = request.files['file']
# Save the file to ./uploads
basepath = os.path.dirname(__file__)
file_path = os.path.join(
basepath, 'uploads', secure_filename(f.filename))
f.save(file_path)
print(file_path)
# Make prediction
preds = model_predict(file_path)
return preds
据我所知,我必须对模型预测的方式或其他方式进行一些更改。如果您知道解决方案,请帮助我。
该图像仅在一个实例中是必需的,因此没有使用它,因此如果有一种方法可以使用我可以虚拟加载图像然后将其删除,那就更好了
Heroku 网址:https://banana-ripeness-detection.herokuapp.com/ Github 网址:https://github.com/Abhayparashar31/brd
当图像上传并发送进行预测时的主要内容和 500 内部错误。
提前致谢