xqh5201314 / YOLOv5-imporve-method

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

YOLOv5-imporve-method

这是本人读研时对YOLOv5的改进 1.安装配置本项目 1)git clone https://github.com/xqh5201314/YOLOv5-imporve-method 2)cd YOLOv5-imporve-method 3)pip install -r requirements.txt 建议从pytorch官网安装pytorch pytorch官网 https://pytorch.org/get-started/previous-versions/ 4)backbone-imporve文件夹是对YOLOv5 backbone的改进; SPPF-imporve文件夹是对SPPF进行改进; neck-imporve文件夹是对neck进行改进; head-imporve 文件夹是对head进行改进; Loss-imporve 是对Loss进行改进; Label_assignment-imporve是对标签分配策略的改进.改进的具体方法见各种的文件夹的readm.md

About

License:GNU Affero General Public License v3.0


Languages

Language:Python 91.6%Language:Jupyter Notebook 6.7%Language:Shell 1.1%Language:Dockerfile 0.6%