From 21a36f151a7dad788910bdbaba0421da4fb0350d Mon Sep 17 00:00:00 2001 From: p73692015 Date: Mon, 1 Jan 2024 15:47:23 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=B3=A8=E9=87=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index 8eba707..136103f 100644 --- a/main.py +++ b/main.py @@ -68,6 +68,7 @@ class mywindow(QtWidgets.QMainWindow, Ui_MainWindow): print('cd {}'.format(self.path1)) print( 'python deploy/python/infer.py --model_dir={} --video_file={} --use_gpu=True'.format(self.model_path, self.Video_fname)) + # 调用GPU os.system('cd {}'.format(self.path1)) os.system( 'python deploy/python/infer.py --model_dir={} --image_dir={} --output_dir=./video_output/{} --threshold=0.3 --use_gpu=True'.format(