GitHub/test_nine
1
0
Fork 0
mirror of https://github.com/luguoyixiazi/test_nine.git synced 2025-12-06 14:52:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

添加docker镜像使用说明

Browse Source
This commit is contained in:
luguoyixiazi
2025-03-21 04:56:31 +08:00
committed by GitHub
parent 513f9dd247
commit 4579be58c9
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@@ -55,11 +55,15 @@ pip install -r requirements.txt
- 运行 `python convert.py`（自行进去修改需要转换的模型，一般是选loss小的） - 运行 `python convert.py`（自行进去修改需要转换的模型，一般是选loss小的）
- paddle模型转换要装paddle2onnx，详情参见https://www.paddlepaddle.org.cn/documentation/docs/guides/advanced/model_to_onnx_cn.html - paddle模型转换要装paddle2onnx，详情参见https://www.paddlepaddle.org.cn/documentation/docs/guides/advanced/model_to_onnx_cn.html
### 5.启动fastapi服务（必须要有训练完成的onnx格式模型） ### 5-a.启动fastapi服务（必须要有训练完成的onnx格式模型）
运行 `python main.py`（默认用的paddle的onnx模型，如果要用resnet18可以自己改注释） 运行 `python main.py`（默认用的paddle的onnx模型，如果要用resnet18可以自己改注释）
由于轨迹问题，可能会出现验证正确但是结果失败，所以建议增加retry次数，训练后的paddle模型正确率在99.9%以上 由于轨迹问题，可能会出现验证正确但是结果失败，所以建议增加retry次数，训练后的paddle模型正确率在99.9%以上
### 5-b.使用docker启动服务
镜像地址为luguoyixiazi/test_nine:25.3.21
运行时只需指定绑定的port即可
### 6.api调用 ### 6.api调用