first commit

GYD_RedCamera.py

@@ -36,7 +36,7 @@ def GYD_RedCameraDeal (root,DetectionCalss,Fine_grainedClass):
 
 
 if __name__ == '__main__':
-    os.environ['CUDA_VISIBLE_DEVICES'] = '4'
+    #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)