PPDetectionPytorch

use pytorch to reimplement paddle detection

简体中文 | English

illustrate

This project aims to:

Learn object detection and PPDetection code
Enable PPDetection model to be used on pytorch
Provide a reference for paddle to pytorch

TODO

Coordinate transformation of target detection in yolov3 inference part
Rotating frame detection model PP-YOLOE-R
Small target detection model PP-YOLOE-SOD
PP-Human v2
PP-Vehicle
Object detection model training part
Multiple target tracking
...

Notice

PPDetectionPytorch is a pytorch implementation based on PaddleDetection and PaddleYOLO.

RECENT UPDATED

2023-05-31 fix issue #1 yolov6 infer update
2023-05-26 Migration model inference part: rt-detr(RT-DETR-R50-m,RT-DETR-R34,RT-DETR-R18)
2023-04-25 Migration model inference part: rt-detr(DETRs Beat YOLOs on Real-time Object Detection)
2023-02-13 Migration model inference part: retinanet
2023-02-11 Migration model inference part: blazeface
2023-02-03 Migration model inference part: ssd
2023-01-20 Migration model inference part: yolov8
2023-01-06 Coordinate transformation of target detection in yolov3 inference part
2022-12-21 Migration model inference part: picodet, ppyoloe, yolox, yolov5, yolov6, yolov7, rtmdet
2022-12-19 Base code migration

Instructions for use

Code path modification

WORK_DIR Constants.WORK_DIR is changed to the absolute path of your own code

Model Inference

Model Weights

Automatically download the corresponding model weights in configs to the ~/.cache/paddle/weights directory during the first inference. This directory is the specified directory for downloading the PPDetection inference model. Model weights will be automatically converted to pytoch weights and saved to the current directory.

Model weight automatic conversion code

Run inference tests

Currently only part of the model weights are converted: The run_picodet_coco_batch function in run_detection_infer.py uses different model_classes for reasoning tests

Inference result output

Output inference results and corresponding network paddle and pytorch model weight names Inference output directory

cycloneboy / ppdetectionpytorch Goto Github PK

ppdetectionpytorch's Introduction

PPDetectionPytorch

illustrate

TODO

Notice

Instructions for use

Code path modification

Model Inference

Model Weights

Run inference tests

Inference result output

refer to

ppdetectionpytorch's People

Contributors

Stargazers

Watchers

Forkers

ppdetectionpytorch's Issues

Recommend Projects

Recommend Topics

Recommend Org