LibreRTDETRl / README.md
Xuban's picture
Model card: add Usage section
f9cf7da verified
|
Raw History Blame Contribute Delete
1.3 kB
metadata
license: apache-2.0
library_name: libreyolo
tags:
  - object-detection
  - rt-detr
  - detr

LibreRTDETRl

RT-DETR with HGNetv2-L backbone, repackaged for the LibreYOLO framework.

Source

Detector: lyuwenyu/RT-DETR. Licensed under the Apache License, Version 2.0. Weights derived from rtdetrv2_hgnetv2_l_6x_coco_from_paddle.pth (lyuwenyu/storage v0.1), which is itself a Paddle→PyTorch conversion of Baidu's official COCO training run.

Backbone: HGNetv2-L (PP-HGNetV2) from PaddlePaddle/PaddleClas. Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. Licensed under the Apache License, Version 2.0.

Modifications

State-dict key remapping and file rename only. Learned parameters are unchanged. Conversion script: weights/convert_rtdetr_hgnetv2_weights.py.

Usage

pip install libreyolo
from libreyolo import LibreYOLO

model = LibreYOLO("LibreRTDETRl.pt")
results = model("image.jpg", save=True)

License

Apache License 2.0. See the LICENSE and NOTICE files in this repository.