diff --git a/aipipeline/projects/bio/run_strided_inference.py b/aipipeline/projects/bio/run_strided_inference.py index 9888afe..25224ae 100644 --- a/aipipeline/projects/bio/run_strided_inference.py +++ b/aipipeline/projects/bio/run_strided_inference.py @@ -387,13 +387,13 @@ def seconds_to_timestamp(seconds): jpg_file.unlink() -def process_videos(video_files, stride, endpoint_url, class_name, version_id, min_confidence, +def process_videos(video_files, stride, endpoint_url, version_id, min_confidence, allowed_classes, class_remap, remove_vignette=False, skip_vss=False): num_cpus = multiprocessing.cpu_count() pool = multiprocessing.Pool(processes=num_cpus) pool.starmap( run_inference, - [(v, stride, endpoint_url, class_name, version_id, min_confidence, allowed_classes, class_remap, remove_vignette, skip_vss) for v in + [(v, stride, endpoint_url, version_id, min_confidence, allowed_classes, class_remap, remove_vignette, skip_vss) for v in video_files], ) pool.close()