From 9270888194e30910bec07bed6abe9dcd09d038bd Mon Sep 17 00:00:00 2001
From: lichen <770918727@qq.com>
Date: Thu, 5 May 2022 11:44:14 +0800
Subject: [PATCH] update

---
 ieemoo-ai-zhanting.py |  2 +-
 requirements.txt      | 13 +++----------
 2 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/ieemoo-ai-zhanting.py b/ieemoo-ai-zhanting.py
index 171740e..58b62f1 100755
--- a/ieemoo-ai-zhanting.py
+++ b/ieemoo-ai-zhanting.py
@@ -21,7 +21,7 @@ app.use_reloader=False
 
 parser = argparse.ArgumentParser()
 #parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5m_finetune/exp4/weights/best.pt', help='model.pt path(s)')
-parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/best.pt', help='model.pt path(s)')
+parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/model/now/best.pt', help='model.pt path(s)')
 parser.add_argument('--source', type=str, default='../module/ieemoo-ai-zhanting/imgs/1.jpg', help='source')  # file/folder, 0 for webcam
 parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
 parser.add_argument('--conf-thres', type=float, default=0.50, help='object confidence threshold')
diff --git a/requirements.txt b/requirements.txt
index 177a0fe..b3b7c3f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,17 +5,10 @@ matplotlib==3.4.1
 numpy==1.20.2
 esdk-obs-python --trusted-host pypi.org
 opencv_python==4.5.1.48
+opencv-contrib-python==4.5.5.64
 Pillow==9.1.0
 scipy==1.6.2
 setuptools==49.6.0
-
---find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.8.0+cu111
---find-links https://download.pytorch.org/whl/torch_stable.html
-torchvision==0.9.0+cu111
---find-links https://download.pytorch.org/whl/torch_stable.html
-torchaudio==0.8.0
-
 coremltools==5.2.0
 onnx==1.7.0
 pandas==1.2.4
@@ -23,6 +16,6 @@ pycocotools==2.0.2
 PyYAML==6.0
 requests==2.25.1
 seaborn==0.11.1
-thop==0.0.31
+thop==0.0.31.post2005241907
 tqdm==4.60.0
-
+ml-collections==0.1.1