use pytorch to reimplement paddle detection
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
- 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
- ...
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
WORK_DIR Constants.WORK_DIR is changed to the absolute path of your own code
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
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
Output inference results and corresponding network paddle and pytorch model weight names Inference output directory