Nioolek / PPYOLOE_pytorch

An unofficial implementation of Pytorch version PP-YOLOE,based on Megvii YOLOX training code.
Apache License 2.0
180 stars 20 forks source link

Any idea of implement on MMDetection codebase? #3

Open cydiachen opened 2 years ago

cydiachen commented 2 years ago

Thx for your excellent job. I am free to provide you with 4 * 3090 gpus to reproduce the PPYoloE results. But most of my work is based on MMDetection codebase. I wish we can reproduce the network on MMDet codebase. Look forward to hear from you.

miemie2013 commented 2 years ago

Thx for your excellent job. I am free to provide you with 4 * 3090 gpus to reproduce the PPYoloE results. But most of my work is based on MMDetection codebase. I wish we can reproduce the network on MMDet codebase. Look forward to hear from you.

Hi! You can train COCO2017 from scratch with my codes https://github.com/miemie2013/miemiedetection

just input these commands:

wget https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams
python tools/convert_weights.py -f exps/ppyoloe/ppyoloe_crn_s_300e_coco.py -c CSPResNetb_s_pretrained.pdparams -oc CSPResNetb_s_pretrained.pth -nc 80 --only_backbone True
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python tools/train.py -f exps/ppyoloe/ppyoloe_crn_s_300e_coco.py -d 8 -b 256 -eb 16 -c CSPResNetb_s_pretrained.pth --fp16

Please refer to https://github.com/miemie2013/miemiedetection/blob/main/docs/README_PPYOLO.md for more details.

Nioolek commented 2 years ago

能通过知乎私发给我你的wechat吗? 知乎链接:https://zhuanlan.zhihu.com/p/505992733

miemie2013 commented 2 years ago

能通过知乎私发给我你的wechat吗? 知乎链接:https://zhuanlan.zhihu.com/p/505992733

I feel sorry to be here.

Nioolek commented 2 years ago

能通过知乎私发给我你的wechat吗? 知乎链接:https://zhuanlan.zhihu.com/p/505992733

I feel sorry to be here.

It doesn't matter I looked at your code, too. It's great

Nioolek commented 2 years ago

@cydiachen I have the idea to reproduce the network on MMDet. I hope we can reproduce it together

cydiachen commented 2 years ago

@cydiachen I have the idea to reproduce the network on MMDet. I hope we can reproduce it together

hello, I have contacted you on Zhihu. Feel free to contact me. I am now profreading your repo and trying on some work of deployment on your excellent work.