From d92cc4fe6d05f04739c97c85b7a588b140583f72 Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Fri, 18 Nov 2022 14:52:36 +0800 Subject: [PATCH 1/2] =?UTF-8?q?=E8=AE=AD=E7=BB=83=E6=A0=B8=E5=BF=83?= =?UTF-8?q?=E6=98=BE=E7=A4=BA=E4=BF=AE=E6=94=B9-2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/controller/AlgorithmController.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py index cc79d94..5cb386c 100644 --- a/app/controller/AlgorithmController.py +++ b/app/controller/AlgorithmController.py @@ -21,7 +21,7 @@ from app.utils.websocket_tool import manager import sys from pathlib import Path -# from pynvml import * +from pynvml import * # FILE = Path(__file__).resolve() # ROOT = FILE.parents[0] # YOLOv5 root directory # if str(ROOT) not in sys.path: From 59cc951d6300fc22b58159215b838b5eb34a7048 Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Fri, 18 Nov 2022 14:54:26 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E8=AE=AD=E7=BB=83=E6=A0=B8=E5=BF=83?= =?UTF-8?q?=E6=98=BE=E7=A4=BA=E4=BF=AE=E6=94=B9-2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/controller/AlgorithmController.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py index 5cb386c..d573149 100644 --- a/app/controller/AlgorithmController.py +++ b/app/controller/AlgorithmController.py @@ -386,7 +386,7 @@ def returnTrainParams(): {"index": 4, "name": "saveModDir", "value": "E:/alg_demo-master/alg_demo/app/yolov5/best.pt", "description": '保存模型路径', "default": "./app/maskrcnn/saved_model/test.pt", "type": "S", 'show': False}, - {"index": 5, "name": "resumeModPath", "value": 'E:/alg_demo-master/alg_demo/app/yolov5/yolov5s.pt', + {"index": 5, "name": "resumeModPath", "value": '/yolov5s.pt', "description": '继续训练路径', "default": '', "type": "S", 'show': False}, {"index": 6, "name": "resumeMod", "value": '', "description": '继续训练模型', "default": '', "type": "E", "items": '',