我正在尝试使用Django创建一个API,并使用POSTMAN对其进行测试。
尝试使用以下代码测试API时出现错误
@api_view(['POST','GET'])
def predict_plant_disease(request):
try:
if request.body:
request_data = request.data["plant_image"]
image_data = request_data.split(';base64,')
image_array,err_msg =image_converter.convert_image(request_data)
if err_msg == None :
model_file = f"{BASE_DIR}/ml_files/cnn_model.pkl"
saved_classifier_model = pickle.load(open(model_file,'rb'))
prediction = saved_classifier_model.predict(image_array)
label_binarizer = pickle.load(open(f"{BASE_DIR}/ml_files/label_transform.pkl",'rb'))
return_data = {
"error" : "0",
"data" : f"{label_binarizer.inverse_transform(prediction)[0]}"
}
else :
return_data = {
"error" : "4",
"message" : f"Error : {err_msg}"
}
else :
return_data = {
"error" : "1",
"message" : "Request Body is empty",
}
except Exception as e:
return_data = {
"error" : "3",
"message" : f"Error : {str(e)}",
}
return HttpResponse(json.dumps(return_data), content_type='application/json; charset=utf-8')
我收到错误消息,因为'InMemoryUploadedFile'对象在if循环的第二行没有属性'split'“。 帮我解决这个错误。
屏幕截图,显示了从POSTMAN API请求结果时的错误。
答案 0 :(得分:1)
从屏幕截图中可以明显看出,您以文件的形式进行输入,而在代码中,图像被接受为base64编码形式。
将输入作为图像的编码base64形式提供。可以使用任何编码网站在线完成此编码。 https://www.base64-image.de/