“错误:'InMemoryUploadedFile'对象没有属性'split'”

时间:2018-12-26 12:15:40

标签: django-rest-framework postman

我正在尝试使用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'“。 帮我解决这个错误。

enter image description here

屏幕截图,显示了从POSTMAN API请求结果时的错误。

1 个答案:

答案 0 :(得分:1)

从屏幕截图中可以明显看出,您以文件的形式进行输入,而在代码中,图像被接受为base64编码形式。

将输入作为图像的编码base64形式提供。可以使用任何编码网站在线完成此编码。 https://www.base64-image.de/