Docker使用及本地Yolov5打包教程

1. Docker的安装

注意:官方也提供了直接Pull Yolov5的渠道:

复制代码
docker pull ultralytics/yolov5

详见:https://hub.docker.com/r/ultralytics/yolov5

--------------------------------------------------以下正文-----------------------------------------------------

建议观看本教学视频:05.安装配置_哔哩哔哩_bilibili

官方下载链接:Docker: Accelerated Container Application Development

注意:需要科学上网注册账号

下载好后如下图:

2. Docker的相关指令学习

docker 常用命令大全_docker常用命令_保护我方胖虎的博客-CSDN博客

【Docker系列】从头学起 Docker------docker run 命令详解_搞什么滚去学习的博客-CSDN博客

3. 案例-打包本地修改好的Yolov5包

3.1 找到本地Yolov5的文件夹:

3.2 创建Dockerfile,注意,没有拓展名。

复制代码
#基于的基础镜像
FROM python:3.9.10
#代码添加到code文件夹
ADD . /usr/src/app/uniform/yolotest
# 设置code文件夹是工作目录
WORKDIR /usr/src/app/uniform/yolotest
# 安装支持
RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ -r requirements.txt

3.3 在Yolov5项目目录的cmd中创建Image

复制代码
docker build -t test .

运行时间可能比较长,运行完成的界面:

在Docker Desktop中就可以看到了:

3.4 Image转为Container

在cmd中运行:

复制代码
docker run -it --gpus all --net=host --ipc=host --privileged=true --name test01 --ulimit core=-1 -v F:/Deep_learning/Dockertest01:/usr/src/app/uniform/data test env LANG=C.UTF-8 /bin/bash 

具体含义可以参考:【Docker系列】从头学起 Docker------docker run 命令详解_搞什么滚去学习的博客-CSDN博客x

现在CMD界面为:

即创建成功,可以试一试调用Python,测试torch的cuda是否可用:

3.5 导入权重文件和测试照片到Container并测试Yolov5

在Docker Desktop中浏览Yolov5文件目录,将想要测试的pt权重文件和照片导入:

直接将pt文件和测试照片拖入即可:

运行yolov5进行检测:

复制代码
python detect.py --weights tomato.pt --source xs_4.jpg --conf-thres 0.7

完成:

相关推荐
Alsn866 小时前
等待学习-学习目录:Docker 容器安全攻防
学习·安全·docker
程序员老赵9 小时前
服务器没有桌面?Docker 跑个 Chrome,浏览器就能远程用
docker·容器·devops
杨浦老苏9 小时前
轻量级Docker仪表板Servedash
运维·docker·监控·群晖·仪表板
正经教主10 小时前
【docker基础】 第八周:容器监控与应用更新策略
运维·docker·容器
kiros_wang10 小时前
Docker 使用完整指南
运维·docker·容器
大鱼>10 小时前
地平线BPU部署实战:YOLOv8在J5/X3上的算法适配与性能优化
算法·yolo·性能优化
正经教主10 小时前
【docker基础】第九周:Docker安全与镜像优化
运维·docker·容器
stsdddd11 小时前
YOLO系列目标检测数据集大全【第二十九期】
yolo·目标检测·目标跟踪
大鱼>11 小时前
YOLO边缘部署深度指南:从YOLOv8n到NPU加速的全链路优化
yolo·aiot
AI棒棒牛11 小时前
第 03 讲《监督学习:数据、标签、Loss与训练循环》
人工智能·学习·yolo·目标检测·yolo26