first commit

GYD_RedCamera.py

@@ -0,0 +1,44 @@
+import os,torch,cv2
+from System_I import  Detection_S,Fine_grained
+from vediotoima import vedio_img
+def GYD_RedCameraDeal (root,DetectionCalss,Fine_grainedClass):
+    CameraLists = os.listdir(root)
+    for CameraIMEI in CameraLists:
+        if not os.path.exists(os.path.join(root+"_Result",CameraIMEI)):
+            os.mkdir(os.path.join(root+"_Result",CameraIMEI))
+        image_videolists = os.listdir(os.path.join(root,CameraIMEI))
+        for image_video in image_videolists:
+            if "JPG" in image_video:
+                pass
+                # print(image_video, " is starting")
+                # image = cv2.imread(os.path.join(root,CameraIMEI,image_video))
+                # DetectionCalss.inference_Red(image,os.path.join(root,CameraIMEI,image_video),Fine_grainedClass)
+                # print(image_video," is ok")
+            else:
+                video_path = os.path.join(root, CameraIMEI,image_video)
+                video_path_out = os.path.join(root + "_Result", CameraIMEI, "RAWImage",image_video.replace(".MP4",""))
+                vedio_img(video_path,video_path_out,25)
+                isExists = os.path.exists(os.path.join(root + "_Result", CameraIMEI, "VideoResult",image_video.replace(".MP4","")))
+                if not isExists:
+                    os.makedirs(os.path.join(root + "_Result", CameraIMEI, "VideoResult",image_video.replace(".MP4","")))
+                img_lists = os.listdir(video_path_out)
+                img_lists.sort()
+       
+                for img in img_lists:
+
+                    image = cv2.imread(os.path.join(video_path_out, img))
+                    DetectionCalss.inference_Red(image, os.path.join(root, CameraIMEI, "VideoResult",image_video.replace(".MP4",""),img),
+                                                 Fine_grainedClass)
+
+
+
+
+
+
+if __name__ == '__main__':
+    os.environ['CUDA_VISIBLE_DEVICES'] = '4'
+    devices = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+    root = "/data1/Hatcher/ftp"
+    Detection1 = Detection_S(devices, root)
+    Fine_grained1 = Fine_grained(devices, root)
+    GYD_RedCameraDeal(root,Detection1,Fine_grained1)