Skip to content

CycloneBoy/PPDetectionPytorch

Repository files navigation

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

refer to