Я использую Django для создания API для моего проекта. Я пытался загрузить модель Keras Conv Net для обработки запросов к конечной точке API, но, похоже, у Keras возникла проблема с загрузкой модели в Django.
Ошибка следует: '_thread._local' object has no attribute 'value'
и указывает на строку, куда я загружаю свою модель Keras. Вот обработчик почтового запроса в моем Django файле просмотра приложений
def post(self, request):
image_serializer = MalariaDiseaseSerializers(data=request.data)
if image_serializer.is_valid():
image_serializer.save()
__model = load_model('trained_models/malaria/malaria.h5')
image = cv2.imread("media/"+str(request.FILES['image']))
image_array = Image.fromarray(image , 'RGB')
resize_img = image_array.resize((50 , 50))
resize_img = np.array(resize_img).reshape(1,50,50,3)/255.
prediction = __model.predict(resize_img)
index = np.argmax(prediction)
self.__response_data['prediction'] = self.__classes[index]
self.__response_data['probability'] = prediction[index]
return Response(self.__response_data, status=status.HTTP_201_CREATED)
return Response(image_serializer.errors, status=status.HTTP_400_BAD_REQUEST)
Может кто-нибудь помочь мне с этим? Спасибо!