yawc

trim_in_directory.py

@@ -0,0 +1,42 @@
+import os
+import random
+from multiprocessing import Pool
+import sys
+sys.path.append('/home/thebears/Seafile/Designs/ML')
+from trim_video import trim_video
+#rtpath = '/srv/ftp/hummingbird/2021'
+rtpath = sys.argv[1]
+
+have_json = set()
+fnames = set()
+for di,_, fns in os.walk(rtpath):
+    for fn in fns:
+        if fn.endswith('.json') and 'trimmed' not in fn:
+            have_json.add(os.path.join(di,fn))
+
+
+files_to_score = have_json
+
+
+
+def try_catch_chunk(jsons):
+    try:
+        if not isinstance(jsons, list):
+            jsons = [jsons]
+
+        for x in jsons:
+            trim_video(x)
+
+    except Exception as e:
+        print(e)
+
+lst = list(files_to_score)
+n = 25
+chunks = [lst[i:i + n] for i in range(0, len(lst), n)]
+# %%
+if __name__ == '__main__':
+    with Pool(8) as p:
+       output = p.map(try_catch_chunk,chunks)
+      #  output = p.map(score_video,chunks)
+
+