Merge pull request #152 from jianzfb/yolo

fix doc
jianzfb · Nov 8, 2024 · eab448b · eab448b
2 parents af8565c + 007ef83
commit eab448b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/antgo/pipeline/functional/mixins/deploy.py b/antgo/pipeline/functional/mixins/deploy.py
@@ -2340,7 +2340,7 @@ def prepare_eagleeye_environment(system_platform, abi_platform, eagleeye_config=
                 if compile_prop_val is not None and compile_prop_val != '':
                     print('Exist rk dependent, dont need download and compile')
                     continue
-                
+
                 root_folder = os.path.abspath(ANTGO_DEPEND_ROOT)
                 os.makedirs(os.path.join(root_folder, 'rk'), exist_ok=True)
                 rk_root_folder = os.path.join(root_folder, 'rk')

diff --git a/doc/算法管线开发10分钟入门.md b/doc/算法管线开发10分钟入门.md
@@ -15,7 +15,10 @@
     # rknn 引擎
     registry.cn-hangzhou.aliyuncs.com/vibstring/rknnconvert:latest
     ```
-
+* 启动开发环境
+    ```
+    docker run -d --shm-size="20G" -p 8080:8080 -e PASSWORD=123 -v /data:/dataset -v /tmp:/tmp -v $(pwd):/workspace -v /var/run/docker.sock:/var/run/docker.sock -v /usr/bin/docker:/usr/bin/docker --gpus all --privileged registry.cn-hangzhou.aliyuncs.com/vibstring/antgo-env-dev /opt/code-server-4.92.2-linux-amd64/bin/code-server --host 0.0.0.0 --auth password
+    ```
 ### 下载配套代码
     incoming
 

diff --git a/docker/model-convert-docker/tnn/convert.py b/docker/model-convert-docker/tnn/convert.py
@@ -47,6 +47,9 @@ def main():
         model_file_name = args.i.split('/')[-1].replace('.onnx', '.opt')
         tnn_model_name = f'{args.o}.{args.version}'
 
+        if not os.path.exists(f'/workspace/{model_file_name}.tnnmodel'):
+            model_file_name = args.i.split('/')[-1].replace('.onnx', '')
+
         os.system(f'mv /workspace/{model_file_name}.tnnmodel /workspace/{tnn_model_name}.tnnmodel')
         os.system(f'mv /workspace/{model_file_name}.tnnproto /workspace/{tnn_model_name}.tnnproto')
         return 0