当前位置: 首页 > news >正文

coco数据集训练nanodet详细流程

github地址

首先要配置环境

conda create -n nanodet python=3.8 -y
conda activate nanodet

确认一下cuda版本

nvcc -V

确认是11.3之后,要安装11.3对应的pytorch版本。
本机装pytorch1.12.1后面运行的时候会报错(torch没有经过cuda编译),于是降级到1.11

conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch

然后进入nanodet文件夹

cd nanodet
pip install -r requirements.txt
python setup.py develop

准备COCO数据集,数据集可以官网下载,也可以代码下载,
代码下载见这篇文章

会得到如下数据集文件夹

在这里插入图片描述
修改nanodet训练的配置文件,在nanodet/config文件夹下,自己选一个模型,
这里选的是nanodet-plus-m_416.yml
需要修改的有下面几个:

save_dir: #训练出的模型保存在哪里
data:train:name: CocoDatasetimg_path: #上面的images文件夹路径ann_path: #上面的annotations文件夹路径/instances_train2017.jsonval:name: CocoDatasetimg_path: #上面的images文件夹路径ann_path: #上面的annotations文件夹路径/instances_val2017.jsondevice:gpu_ids: [0]  #想用CPU的这里写-1workers_per_gpu: 10batchsize_per_gpu: 16  #需要根据实际需要修改total_epochs: 50  #训练多少个epoch
val_intervals: 10  #每10个epoch做一次evaluation

设置参数
在pycharm的Run->Edit Configurations
Parameters里填上面的配置文件nanodet-plus-m_416.yml所在的完整路径

开始训练,训练完成后模型model_last.ckpt会保存在save_dir文件夹下。

来测试一下这个模型,
选择nanodet/demo/demo.py
设置参数
Parameters里填上

--config ../config/nanodet-plus-m_416.yml --model save_dir路径/model_last.ckpt --path 测试图片路径/people.jpg --save_result image

这是训练50个epoch的结果。
在这里插入图片描述

http://www.lryc.cn/news/17028.html

相关文章:

  • 关于Kubernetes不兼容Docker
  • MySql_基础篇_从下载mysq到数据库操作
  • 当我尝试问了chatGPT几个问题之后,我感到了危机......
  • 如何生成全局唯一ID
  • 基于node vue的电商系统 mongodb express框架
  • 【03-JVM面试专题-JVM中你知道的类加载器有哪些?作用是什么?分别用来加载什么文件?什么内容的呢?】
  • 面试纪要FFFFF
  • gma 1.1.3 (2023.02.22) 更新日志
  • 现有安全技术
  • Qt QMAKE_MSC_VER
  • 如何通俗地理解原码、反码和补码
  • 在vite vue3 前端架构中,切换环境,切换项目的架构设计方案
  • 华为OD机试真题Python实现【跳格子】真题+解题思路+代码(20222023)
  • Python 入门之文件和异常处理
  • 操作系统作业
  • 【计算机网络 -- 期末复习】
  • 三、(补充)接口是对类的一部分行为的抽象
  • CIMCAI intellgent ship product applied by world top3 shipcompany
  • 媒体见面会怎么做?
  • Nginx面试题一步到位
  • 华为OD机试真题 用 C++ 实现 - 括号检查
  • Windows下SecureCRT的下载、安装、使用、配置【Telnet/ssh/Serial】
  • Git 分支操作
  • 【面试题】TCP如何保证传输可靠性?TCP流量控制实现、拥塞控制、ARQ协议、停止等待ARQ、连续ARQ
  • MySQL一隐式转换
  • 风光并网对电网电能质量影响的matlab/simulink仿真建模
  • 浅谈Spring循环依赖
  • 华为OD机试题 - 拼接 URL(JavaScript)| 包含代码编写思路
  • 【FFMPEG】Filtering Introduction[翻译/举例]
  • 什么是IP65?仅仅是防水等级吗?看完本文直呼666!