output
Browse files
main.py
CHANGED
@@ -39,7 +39,7 @@ def check_extension(video):
|
|
39 |
file_name = split_tup[0]
|
40 |
file_extension = split_tup[1]
|
41 |
print(file_extension)
|
42 |
-
|
43 |
if file_extension is not ".mp4":
|
44 |
clip = moviepy.VideoFileClip(video)
|
45 |
|
@@ -88,11 +88,11 @@ def show_tracking(video_content):
|
|
88 |
# Go through frames and write them
|
89 |
for frame_track in video_track:
|
90 |
result_track = frame_track[0].plot() # plot a BGR numpy array of predictions
|
91 |
-
|
|
|
92 |
print("[INFO] Done with frames")
|
93 |
#print(type(result_pose)) numpy ndarray
|
94 |
-
|
95 |
-
|
96 |
out_track.release()
|
97 |
|
98 |
video.release()
|
|
|
39 |
file_name = split_tup[0]
|
40 |
file_extension = split_tup[1]
|
41 |
print(file_extension)
|
42 |
+
|
43 |
if file_extension is not ".mp4":
|
44 |
clip = moviepy.VideoFileClip(video)
|
45 |
|
|
|
88 |
# Go through frames and write them
|
89 |
for frame_track in video_track:
|
90 |
result_track = frame_track[0].plot() # plot a BGR numpy array of predictions
|
91 |
+
out_track.write(result_track)
|
92 |
+
|
93 |
print("[INFO] Done with frames")
|
94 |
#print(type(result_pose)) numpy ndarray
|
95 |
+
|
|
|
96 |
out_track.release()
|
97 |
|
98 |
video.release()
|