基于 chatgpt 产生的代码, 部署ymir模型到云端
- 执行以下命令后, 浏览器打开 127.0.0.1:15500 或 <服务器ip>:15500 可以看到 图片上传页面
docker pull youdaoyzbx/ymir-executor:flask-api
docker run -it --rm --gpus all --ipc host -p 15500:15500 youdaoyzbx/ymir-executor:flask-api
- 上传一张包含狗的图片,点击预测可得如下结果
- 基于镜像
youdaoyzbx/ymir-executor:ymir2.1.0-yolov5-v7.0-cu111-tmi
训练好模型,并下载解压到本地,将其最好的模型重命名
mv bee8ab5263660fa015cd1a8199c3974cbacc1c62 bee8ab5263660fa015cd1a8199c3974cbacc1c62.tar
tar -xf bee8ab5263660fa015cd1a8199c3974cbacc1c62.tar
mv yolov5_best/best.pt dog.pt
- 在
server.py
中配置好class_names
与weigths
class_names = ['dog']
cmd = f'python3 detect.py --source {time_stamp}.jpg --weights dog.pt --save-txt --save-conf --exist-ok'
subprocess.run(cmd.split(), check=True)
docker build -t youdaoyzbx/ymir-executor:my-flask-api -f dockerfile .
docker run -it --rm --gpus all --ipc host -p 15500:15500 youdaoyzbx/ymir-executor:my-flask-api
浏览器打开127.0.0.1:15500 或 <服务器ip>:15500