Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def resize_frame(frame, scale_percent):
|
|
23 |
return resized
|
24 |
|
25 |
@spaces.GPU
|
26 |
-
def process_video(video_file, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes, verbose, progress
|
27 |
# Ensure classes is a list of integers
|
28 |
classes = [int(x) for x in classes.split(',') if x.strip().isdigit()] if classes else None
|
29 |
|
@@ -93,7 +93,7 @@ def process_video(video_file, scale_percent, line_start_x, line_start_y, line_en
|
|
93 |
|
94 |
# Update progress
|
95 |
processed_frames += 1
|
96 |
-
progress
|
97 |
|
98 |
cap.release()
|
99 |
video_writer.release()
|
@@ -129,8 +129,7 @@ def preview_line(video_file, scale_percent, line_start_x, line_start_y, line_end
|
|
129 |
return preview_path
|
130 |
|
131 |
def gradio_app(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes_to_track, verbose):
|
132 |
-
|
133 |
-
output_path = process_video(video.name, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, int(line_thickness), draw_tracks, view_img, view_in_counts, view_out_counts, int(track_thickness), int(region_thickness), line_dist_thresh, persist, conf, iou, classes_to_track, verbose, progress)
|
134 |
return output_path, output_path
|
135 |
|
136 |
def update_preview(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness):
|
|
|
23 |
return resized
|
24 |
|
25 |
@spaces.GPU
|
26 |
+
def process_video(video_file, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes, verbose, progress):
|
27 |
# Ensure classes is a list of integers
|
28 |
classes = [int(x) for x in classes.split(',') if x.strip().isdigit()] if classes else None
|
29 |
|
|
|
93 |
|
94 |
# Update progress
|
95 |
processed_frames += 1
|
96 |
+
progress((processed_frames / total_frames) * 100)
|
97 |
|
98 |
cap.release()
|
99 |
video_writer.release()
|
|
|
129 |
return preview_path
|
130 |
|
131 |
def gradio_app(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes_to_track, verbose):
|
132 |
+
output_path = process_video(video.name, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, int(line_thickness), draw_tracks, view_img, view_in_counts, view_out_counts, int(track_thickness), int(region_thickness), line_dist_thresh, persist, conf, iou, classes_to_track, verbose, gr.Progress())
|
|
|
133 |
return output_path, output_path
|
134 |
|
135 |
def update_preview(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness):
|