first commit

Hatcher_is_cool.py

@@ -0,0 +1,58 @@
+'''
+author ：  Hatcher W
+Email： hcwang@semi.ac.cn
+use: 递归一个文件下面所有的视频和图像的路径
+input:一个文件夹目录
+output：两个json文件，分别存储视频和图像的路径
+'''
+import os
+import shutil
+import  json
+
+vediofile = []
+imagefile = []
+vedio_format = [".MP4" , ".mp4" , ".MOV" , ".mov"]
+image_format = [".JPG" , ".JPEG" , ".jpg"]
+def eachfile(filepath):
+    pathdir = os.listdir(filepath)
+    for s in pathdir:
+        newdir = os.path.join(filepath,s)#将文件名加到当前文件路径后面
+        if os.path.isfile(newdir):  #判断是不是文件
+            print(os.path.splitext(newdir)[1])
+            if os.path.splitext(newdir)[1] in vedio_format:
+                vediofile.append(newdir)
+            elif os.path.splitext(newdir)[1]in image_format:
+                imagefile.append(newdir)
+            elif os.path.isdir(newdir):
+                eachfile(newdir)
+        else:
+            pathdir_1 = os.listdir(newdir)   #取出下一层的路径
+            for s_1 in pathdir_1:           #遍历下一层的路径
+                newdir_1 = os.path.join(newdir,s_1)     #取出其中一个路径
+                if os.path.isfile(newdir_1):  # 判断是不是文件
+                    print(os.path.splitext(newdir_1)[1])
+                    if os.path.splitext(newdir_1)[1] in vedio_format:
+                        vediofile.append(newdir_1)
+                    elif os.path.splitext(newdir_1)[1] in image_format:
+                        imagefile.append(newdir_1)
+                    elif os.path.isdir(newdir_1):
+                        eachfile(newdir_1)
+                else:eachfile(newdir_1)
+    return vediofile,imagefile
+
+fp = "/data1/Hatcher/DATA/Hat_Raw_Data"
+
+vedio_f,image_f = eachfile(fp)
+print(vedio_f)
+print(image_f)
+with open ("./vedioroot.json",'w') as f:
+    json.dump(vedio_f,f)
+    f.close()
+with open ("./imageroot.json",'w') as g:
+    json.dump(image_f,g)
+    g.close()
+for i in range(len(vedio_f)):
+    pcm_file = vedio_f[i]
+    path_0 = os.path.split(vedio_f[i])[0]
+    path_1 = os.path.split(vedio_f[i])[1]
+    #print(path_1)