Развертывание модели Keras в django - PullRequest
0 голосов
/ 29 апреля 2020

Я использую 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)

Может кто-нибудь помочь мне с этим? Спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...