Skip to content

在别人的代码中看到一些晦涩难懂的函数过程,因此结合别人的代码,对yolov8按照自己理解复写。

Notifications You must be signed in to change notification settings

SpringCCC/yolov8

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

目标step:

  • 自己的理解复现yolov8
    • 参考ultra数据加载方式,使用队列来加快mosaic数据增强
  • 替换损失函数为yolox
  • 替换模型为yolox
  • 类似插件方式,自定义选择模型和损失函数:yolov8还是yolox

数据格式为: 一个jpg对应一个txt文件 txt文件中每一行代表一个目标 每个目标: cls, cx, cy, w, h 且归一化为0-1

训练时,给定一个txt文件,txt文件每一行代表一个jpg_file,替换掉后缀名为txt后,为txt路径 input_shape: (h, w)

#训练图像处理流程

  • Image读取图像
  • 扭曲/水平反转图像
  • HSV色域变换
  • mosaic数据增强
  • 归一化到0-1

基本原则: 所有图像都是rgb,从读取之后就是这样 图像resize需要pad时,pad值默认为128

模型的输出为y, x x是经过FPN+PAN之后的输出 y是x经过解码后的输出 dbox, cls

About

在别人的代码中看到一些晦涩难懂的函数过程,因此结合别人的代码,对yolov8按照自己理解复写。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages