Блог

Главная - Вопросы по программированию - как правильно использовать модель cnn lstm для видео

как правильно использовать модель cnn lstm для видео

#python #deep-learning #conv-neural-network #lstm

Вопрос:

Я пытаюсь протестировать свою модель на видео для прогнозирования.

Я хочу сделать прогноз, используя свою cnn(alexnet) lstm модель на видео, которое у меня есть, но когда оно запускается, прогноз всегда показывает результат 0, даже если я продолжаю изменять свое видео, чтобы увидеть другой результат, но прогноз не изменился

Вот мой код:

 model_path = 'Model/AlexNet_LSTM_10Frame_Cropped.h5'
model = models.load_model(model_path)
vid = cv2.VideoCapture("Data Fix/Data16_133.mp4")

while(vid.isOpened()):
    ret, frame = vid.read()
    vid.set(3, 480)
    vid.set(4, 240)
    start = time.time()
    if ret == True:
        total_frame  = 1
        draw = frame.copy()
        draw = cv2.cvtColor(draw, cv2.COLOR_BGR2RGB)
        scale_percent = 20 # percent of original size
        width = 224
        height = width
        dim = (width, height)
        frame_set = cv2.resize(draw, dim, interpolation = cv2.INTER_AREA)
        frame_set=np.arange(10*width*height*3).reshape(10,width, height, 3)
        frame_set.reshape(10, width, height, 3).shape
        frame_set = np.expand_dims(frame_set, axis=0)
        result=model.predict_on_batch(frame_set)

        cv2.imshow('Result', result)
        print(result)
        if cv2.waitKey(1) amp; 0xFF == ord('q'):
            break            

vid.release()
cv2.destroyAllWindows()

кто-нибудь знает, как мне внедрить свою модель cnn lstm в видео?была ли моя ошибка в программе, которую я сделал, или моя ошибка при создании модели?

в настоящее время я пытаюсь в этом уроке сделать модель и способ ввода пути к набору данных такими же, как в этом уроке, разница в том, что я не использовал MobileNet метод обучения передаче, я изменил его с помощью AlexNet модели, но не метода обучения передаче.

любой ответ был бы признателен

Огромное спасибо!

Метки: