Add example
This commit is contained in:
Hell13Cat
2022-06-30 22:31:06 +03:00
parent ae846f2e3e
commit c9a7b13c5a
9 changed files with 75 additions and 2 deletions
+1 -1
View File
@@ -14,7 +14,7 @@ def save_frame(screen_folder, secs, vid):
return save_name
model = predict.load_model('./data/model/nsfw_mobilenet2.224x224.h5')
file_path = "./data/vids/5.mp4"
file_path = "./data/vids/example.mp4"
vid = cv2.VideoCapture( file_path )
height = vid.get(cv2.CAP_PROP_FRAME_HEIGHT)
width = vid.get(cv2.CAP_PROP_FRAME_WIDTH)