yolov5制作docker镜像并在容器中使用

说明：本教程为将yolov5制作为docker镜像，通过容器实现yolov5的调用。

一、建立包含yolov5所需库的dockerfile，基于普通深度学习dockerfile修改，具体dockerfile文件内容如下：

# 继承PyTorch的cuda镜像
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel

LABEL maintainer = "787255035@qq.com"
LABEL version = "0.2"
LABEL description = "prepare deep learning environment"

# 指定docker镜像中，默认的工作路径是/PD
WORKDIR /home/lwf/PD/yolov5

COPY ./yolov5/ /home/lwf/PD/yolov5/

RUN rm /etc/apt/sources.list.d/cuda.list \
   && rm /etc/apt/sources.list.d/nvidia-ml.list \
   && apt-get update \
   && pip config set global.index-url https://pypi.douban.com/simple/ \
   && apt install -y libgl1-mesa-glx \
   && apt-get install -y libglib2.0-dev \
   && pip install gitpython>=3.1.30 \
   && pip install pandas==2.0.0 \
   && pip install seaborn==0.12.2 \
   matplotlib>=3.3 \
   numpy>=1.18.5 \
   opencv-python==4.6.0.66 \
   #    opencv-python-headless \
   Pillow>=7.1. \
   psutil # system resources \
   PyYAML>=5.3.1 \
   requests>=2.23.0 \
   scipy>=1.4.1 \
   thop>=0.1.1 # FLOPs computation \
   torch>=1.7.0 # see https://pytorch.org/get-started/locally (recommended) \
   torchvision>=0.8.1 \
   tqdm>=4.64.0 \
   setuptools>=65.5.1 # Snyk vulnerability fix \
        prefetch_generator

以上文件中COPY ./yolov5/ /home/lwf/PD/yolov5/实现将本地dockerfile所在当前文件夹下的yolov5工程拷贝至镜像中，为后期实现数据与代码互相隔离。

二、在当前文件夹下开启终端，运行以下指令制作镜像：

docker build -t lwf/pd:1.0 .

需要对库进行下载，等待镜像制作完成，使用docker images指令查看是否制作成功。

三、创建容器，对需要宿主机链接的文件进行链接：

docker run -it --gpus all -v /home/nvidia/Liuwf/docker_work/PestDetection/input_img:/home/lwf/PD/yolov5/input_img -v /home/nvidia/Liuwf/docker_work/PestDetection/output_path:/home/lwf/PD/yolov5/output_path -v /home/nvidia/Liuwf/docker_work/PestDetection/weights:/home/lwf/PD/yolov5/weights lwf/pd:1.0

四、在相应文件夹下放置对应文件后，指令运行代码：

python detect.py --save-txt  #需要保存坐标文件传入此参数--save-txt

对应链接宿主机目录下可查看工程运行后文件。

五、补充：已有容器查看指令

docker ps -a

yolov5制作docker镜像并在容器中使用相关推荐

Ubuntu18.04下安装NVIDIA显卡驱动、docker、nvidia-docker；容器中编译安装opencv-4.4.0与darknet-yolov4并完成测试；容器封装镜像转移。2022
记录一下第一次在CSDN发博客,欢迎大家光临~ 文章目录前言一.宿主机配置 1.安装Ubunntu18.04 64位系统 2.为宿主机系统更换国内软件源Ubuntu 官方源服务器在欧洲,国内访问很 ...
docker hub push_如何制作Docker镜像（image）？
制作Docker镜像一般有2种方法: 使用hub仓库中已有的环境,安装自己使用的软件环境后完成image创建通过Dockerfile,完成镜像image的创建下面通过展示具体操作方法: 第一种:使 ...
龙芯Fedora21平台制作docker镜像，并且解决vi乱码问题
http://ask.loongnix.org/?/article/81 实验环境本文的实验都是在龙芯3A3000机器上. 操作系统是loongnix(Fedora21)20170726版本, ...
用GitHub Actions制作Docker镜像
欢迎访问我的GitHub 这里分类和汇总了欣宸的全部原创(含配套源码):https://github.com/zq2599/blog_demos 关于GitHub Actions GitHub Act ...
jar包制作Docker镜像
一.编写Dokcerfile ###FROM 指定该docker镜像运行环境(必须有此docker镜像,此处选择jdk:8) FROM ubuntu_java8:v1 ###将准备的配置文件添加到容器 ...
在jetson xavier nx上制作docker镜像
概览: 一.docker简介二.在jetson xavier nx上制作docker镜像一.docker简介用途:对于一项工程,不同的机器都需要单独为其配置环境,且有时配环境是一件相当麻烦的事情 ...
Java Spring Boot 2.0 实战之制作Docker镜像并推送到Docker Hub和阿里云仓库
内容摘要:大规模集群快速部署Java应用,需要制作Docker镜像,本次课程详细介绍如何制作Java程序的Docker镜像文件,深入解析DockerFile核心参数,以及实践演练把我们制作的Docke ...
【Docker】备份Docker镜像image、容器container、数据volume
文章目录镜像image备份容器container备份数据卷volume 基本操作数据卷备份 reference 本意上想要了解一下 Docker 容器中 Volume 的备份,毕竟重要的数据都 ...
mysql sshd_制作Docker镜像之mysqlkeepalivedsshd
制作Docker镜像之mysql&keepalived&sshd mysql docker keepalived ssh 注:该实验采用的docker版本:Docker version ...

yolov5制作docker镜像并在容器中使用

yolov5制作docker镜像并在容器中使用相关推荐

最新文章

热门文章