上传

Vision/bag_collection.py

@@ -31,7 +31,7 @@ class DetectionBag:
         model_path = ''.join([os.getcwd(), '/Vision/model/pt/bag_collection.pt'])
         self.camera_rvc = camera_pe()
         self.imgsz = 640
-        self.cuda = 'cpu'
+        self.cuda = '0'
         self.conf = 0.40
         self.iou = 0.45
         self.model = AutoBackend(model_path, device=torch.device(self.cuda))
@@ -82,7 +82,7 @@ class DetectionBag:
                 if Bag==True:
                     if get_disk_space(path=os.getcwd()) < 15:  # 内存小于15G,停止保存数据
                         save_img_point = 0
-                        print('系统内存不足，无法保存数据')
+                        print('硬盘空间不足，无法保存数据')
                     else:
                         save_path = ''.join([os.getcwd(), '/Vision/model/data/',
                                              time.strftime('%Y_%m_%d_%H_%M_%S', time.localtime(time.time()))])