瀏覽代碼

模型更改

xujiayue 1 年之前
父節點
當前提交
bbdd49428f
共有 4 個文件被更改,包括 5 次插入4 次删除
  1. 1 1
      .idea/misc.xml
  2. 1 1
      .idea/yl-ocr-layout.iml
  3. 2 1
      Dockerfile
  4. 1 1
      server.py

+ 1 - 1
.idea/misc.xml

@@ -1,4 +1,4 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (py38)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (torch13)" project-jdk-type="Python SDK" />
 </project>

+ 1 - 1
.idea/yl-ocr-layout.iml

@@ -2,7 +2,7 @@
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$" />
-    <orderEntry type="jdk" jdkName="Python 3.8 (py38)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.8 (torch13)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
   <component name="PyDocumentationSettings">

+ 2 - 1
Dockerfile

@@ -103,7 +103,8 @@ RUN git clone https://gitee.com/monkeycc/yolov5.git
 # RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/best.pt
 # RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/best2.pt
 #RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/8-14_layout_model.pt
-RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/0822yili.pt
+#RUN wget --user=sxkj --password='sx' ftp://192.168.199.208/0822yili.pt
+RUN wget  ftp://192.168.199.31/11-17.pt
 
 
 ADD . /workspace

+ 1 - 1
server.py

@@ -18,7 +18,7 @@ YOLO_DIR = '/workspace/yolov5'
 # WEIGHTS = '/workspace/yili.pt'
 # WEIGHTS = '/workspace/best2.pt'
 
-WEIGHTS = '/workspace/0822yili.pt'
+WEIGHTS = '/workspace/11-17.pt'
 
 logger = logging.getLogger('log')
 logger.setLevel(logging.DEBUG)