• Stars
    star
    282
  • Rank 146,549 (Top 3 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created almost 3 years ago
  • Updated 9 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Pytorch and ncnn implementation of PPYOLOE、YOLOX、PPYOLO、PPYOLOv2、PicoDet and so on.

简体中文 | English

miemiedetection

概述

miemiedetection是咩酱基于YOLOX进行二次开发的个人检测库(使用的深度学习框架为pytorch),实现了可变形卷积DCNv2、Matrix NMS等高难度算子,支持单机单卡、单机多卡、多机多卡训练模式(多卡训练模式建议使用Linux系统),支持Windows、Linux系统,以咩酱的名字命名。miemiedetection是一个不需要安装的检测库用户可以直接更改其代码改变执行逻辑,所见即所得!所以往miemiedetection里加入新的算法是一件很容易的事情(可以参考PPYOLO的写法往miemiedetection里加入新的算法)。得益于YOLOX的优秀架构,miemiedetection里的算法训练速度都非常快,数据读取不再是训练速度的瓶颈!目前miemiedetection支持YOLOX、PPYOLO、PPYOLOv2、PPYOLOE、SOLOv2等算法,预计未来会加入更多算法,所以请大家点个star吧!

安装依赖

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

torch版本建议1.10.1+cu102或者更高;torchvision版本建议0.11.2+cu102或者更高。

支持的算法

Updates!!

  • 【2023/04/17】 加入了PicoDet算法(基于PaddleDetection-release-2.6版本的代码复现)!实测可以复现官方的精度!转换权重、复现COCO上的精度相关命令见readme_picodet.txt,训练日志见train_coco/picodet_s_416_coco_lcnet_4gpu.txt
  • 【2022/08/03】 PPYOLOv2、PPYOLO算法支持导出到NCNN!详情请参考PPYOLO 文档的“NCNN”小节。
  • 【2022/06/22】 PPYOLOE算法支持导出到NCNN!详情请参考PPYOLOE 文档的“NCNN”小节。
  • 【2022/05/15】 加入了PPYOLOE算法(基于PaddleDetection-release-2.4版本的代码复现)!实测可以复现官方的精度!命令请参考PPYOLOE 文档的“训练COCO数据集”小节。训练日志见train_coco/ppyoloe_s_8gpu.txt

友情链接

  • miemieGAN miemieGAN是咩酱个人开发与维护的图像生成库,以咩酱的名字命名,实现了stylegan2ada等算法,目前文档完善中,欢迎大家试玩。

传送门

算法1群:645796480(人已满)

算法2群:894642886

粉丝群:704991252

关于仓库的疑问尽量在Issues上提,避免重复解答。

B站不定时女装: _糖蜜

知乎不定时谢邀、写文章: 咩咩2013

西瓜视频: 咩咩2013

微信:wer186259

本人微信公众号:miemie_2013

技术博客:https://blog.csdn.net/qq_27311165

AIStudio主页:asasasaaawws

欢迎在GitHub或上面的平台关注我(求粉)~

打赏

如果你觉得这个仓库对你很有帮助,可以给我打钱↓

Example 0

咩酱爱你哟!

引用

miemiedetection

YOLOX

PaddleDetection

PPYOLOE paper

 @article{yolox2021,
  title={YOLOX: Exceeding YOLO Series in 2021},
  author={Ge, Zheng and Liu, Songtao and Wang, Feng and Li, Zeming and Sun, Jian},
  journal={arXiv preprint arXiv:2107.08430},
  year={2021}
}
@article{huang2021pp,
  title={PP-YOLOv2: A Practical Object Detector},
  author={Huang, Xin and Wang, Xinxin and Lv, Wenyu and Bai, Xiaying and Long, Xiang and Deng, Kaipeng and Dang, Qingqing and Han, Shumin and Liu, Qiwen and Hu, Xiaoguang and others},
  journal={arXiv preprint arXiv:2104.10419},
  year={2021}
}
@misc{long2020ppyolo,
title={PP-YOLO: An Effective and Efficient Implementation of Object Detector},
author={Xiang Long and Kaipeng Deng and Guanzhong Wang and Yang Zhang and Qingqing Dang and Yuan Gao and Hui Shen and Jianguo Ren and Shumin Han and Errui Ding and Shilei Wen},
year={2020},
eprint={2007.12099},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
@misc{ppdet2019,
title={PaddleDetection, Object detection and instance segmentation toolkit based on PaddlePaddle.},
author={PaddlePaddle Authors},
howpublished = {\url{https://github.com/PaddlePaddle/PaddleDetection}},
year={2019}
}

More Repositories

1

Keras-YOLOv4

yolov4 42.0% mAP.ppyolo 45.1% mAP.
Python
446
star
2

Pytorch-PPYOLO

ppyolo in pytorch. 44.8% box mAP.
Python
105
star
3

Pytorch-DIOU-YOLOv3

Pytorch复现YOLOv3,使用最新的DIOU loss训练
Python
69
star
4

Keras-DIOU-YOLOv3

Keras上700行代码复现YOLOv3!使用DIOU loss。支持将模型导出为pytorch模型。
Python
60
star
5

Paddle-YOLOv4

Paddle-YOLOv4,supports training, at least 41.1% mAP.支持训练,至少41.1%mAP。少数的给出精度的复现。
Python
39
star
6

miemieGAN

Pytorch implementation of StyleGAN2ADA
Python
31
star
7

Keras-PPYOLO-YOLOv4

45.1% mAP, Keras impl of PPYOLO and YOLOv4.
Python
29
star
8

Paddle-SOLOv2

Paddle2.0动态图版SOLOv2, 38.8%mask AP.
Python
22
star
9

Keras-SOLO

Keras-SOLO(Segmenting Objects by Locations)
Python
21
star
10

Keras-DCNv2

Deformable Convolutional Networks v2 with Keras and Tensorflow1.x
Python
19
star
11

Pytorch-YOLOv4

supports training, at least 41.1% mAP.支持训练,至少41.1%mAP。少数的给出精度的复现。
Python
18
star
12

Pure_Python_Deep_Learning

纯Python实现的深度学习框架,帮助你理解底层细节斩获offer
Python
17
star
13

PaddlePaddle_yolact

yolact完美复刻版(paddle实现),飞桨论文复现挑战赛参赛作品
Python
14
star
14

Paddle-DIOU-YOLOv3

Paddle上实现yolov3算法
Python
12
star
15

miemienet

miemienet is a C++ AI deep learning inference framework.Supports PPYOLOE、PICODET.
C++
12
star
16

Paddle-PPYOLO

Paddle2.0动态图版PPYOLO, 45.1%box AP.
Python
11
star
17

ncnn_ppyolov2

ncnn impl of ppyolov2 and ppyolo and ppyoloe.
C++
8
star
18

ncnn

ncnn implementation of StyleGAN2ADA and StyleGAN3.
C++
8
star
19

PaddleDetection-SOLOv2

SOLOv2,实时实例分割
Python
8
star
20

keras_CartoonGAN

CartoonGAN移植到keras,一个支持穿越次元壁的GAN。
Python
7
star
21

Pytorch-FCOS

Python
7
star
22

Paddle-YOLOX

Python
3
star
23

miemieLabels

A Object Detection label tools.一个目标检测标注工具。
Python
3
star
24

PaddleDetection-release-2.6

Python
2
star
25

miemieyolo

Jupyter Notebook
2
star
26

ppgan

PaddlePaddle implementation of Stylegan2ada.
Python
2
star
27

ncnn-android-ppyoloe

C++
1
star
28

ncnn3

C++
1
star
29

Paddle-FCOS

FCOS实时版,速度超过YOLOv4
Python
1
star
30

miemiecourse

Python
1
star