search

pedrofrodenas / Violence-Detection-CNN-LSTM

Violence Detection tutorial using pre-trained CNN and LSTM
GNU General Public License v3.0
29 stars 16 forks source link

Test #1

Open molstmare opened 5 years ago

molstmare commented 5 years ago

How can i test with the model obtained if i give him a video and see if is fight or not fight ?

CainXIII commented 4 years ago

How can i test with the model obtained if i give him a video and see if is fight or not fight ?

video = get_transfer_values('data', 'videoplayback_no.avi') video = video.reshape((1, video.shape[0], video.shape[1])) label = np.argmax(model.predict(video))

you can test the model like this. But this's a classification model, not a detection