Skip to main content

rtdetr

Real-time object detection you can ship: PyTorch training, OpenVINO inference, 100% Apache-2.0 — code and pretrained weights alike.

detections on a street scene

Install

pip install rtdetr              # inference
pip install "rtdetr[train]"     # + training

Detect

from rtdetr import RTDETR

model = RTDETR("rtdetr-r18")          # COCO weights, downloaded on first use
r = model("bus.jpg", conf=0.5)[0]

r.boxes.xyxy, r.boxes.conf, r.boxes.cls   # plain numpy
r.names[int(r.boxes.cls[0])]              # "person"
r.plot(); r.save(); r.show()

Point it at an image, a folder, a glob, a video, an RTSP stream, a webcam index, or a numpy array. Long sources stream:

for r in model.predict(0, stream=True, show=True):   # webcam, q or Esc quits
    print(r.boxes.xyxyn)

model.track("clip.mp4")        # adds r.boxes.id

Label, train and deploy in a browser

pip install "rtdetr[studio]"
rtdetr studio                                  # http://127.0.0.1:8080

the studio

Upload a dataset, train, watch the curve, download the weights and the OpenVINO IR. For drawing the boxes in the first place:

rtdetr label source=images/ names=can,bottle   # auto-label, then fix

Train

model = RTDETR("rtdetr-r18")
model.train(data="data.yaml", epochs=100, imgsz=640, batch=8, device=0)
model.val(data="data.yaml").box.map50
model.export(format="openvino", half=True)     # IR + labels.txt
# data.yaml
path: /data/cans
train: images/train
val: images/val
names: {0: can, 1: bottle}

Command line

rtdetr predict model=rtdetr-r18 source=bus.jpg conf=0.5
rtdetr train   model=rtdetr-r18 data=data.yaml epochs=100
rtdetr val     model=best.pt data=data.yaml
rtdetr export  model=best.pt format=openvino half=true
rtdetr label   source=images/ names=can,bottle
rtdetr studio  port=8080

Models

name params COCO AP notes
rtdetr-r18 20M 46.4 fastest
rtdetr-r34 31M 48.9
rtdetr-r50 43M 53.1 most accurate

Docs

  • Using the model — sources, results, tracking, saving
  • Labelling — the browser tool, auto-labelling
  • Studio — the web app: datasets, jobs, artefacts, API
  • Training — datasets, validation, export, CLI
  • Weights — the mirror, building it, offline use
  • Performance — measured speeds and how to improve them
  • Design — architecture and provenance
  • Development — tests, releases

한국어

from rtdetr import RTDETR

model = RTDETR("rtdetr-r18")               # COCO 사전학습 가중치 자동 다운로드
model("bus.jpg", conf=0.5)[0].save()       # 결과 이미지 저장
model.train(data="data.yaml", epochs=100)  # 내 데이터로 학습
model.export(format="openvino", half=True) # 배포용 IR + labels.txt

가중치 캐시는 ~/.rtdetr/, 사내 미러는 RTDETR_ASSETS_URL 환경변수로 지정합니다.

License

Apache-2.0 — see LICENSE and NOTICE.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

rtdetr-0.5.0.tar.gz (79.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

rtdetr-0.5.0-py3-none-any.whl (73.7 kB view details)

Uploaded Python 3

File details

Details for the file rtdetr-0.5.0.tar.gz.

File metadata

  • Download URL: rtdetr-0.5.0.tar.gz
  • Upload date:
  • Size: 79.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for rtdetr-0.5.0.tar.gz
Algorithm Hash digest
SHA256 66609ef6e16363a1df6605498d85f3af325121c287beedbcb20d8fcf6ea7a60e
MD5 f6866a329e5539225db4467a28e373e8
BLAKE2b-256 0490e6b259c580a8333d8e894310940924087a926165c303e50c955477b9fb78

See more details on using hashes here.

Provenance

The following attestation bundles were made for rtdetr-0.5.0.tar.gz:

Publisher: publish.yml on leeyunjai82/rtdetr

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file rtdetr-0.5.0-py3-none-any.whl.

File metadata

  • Download URL: rtdetr-0.5.0-py3-none-any.whl
  • Upload date:
  • Size: 73.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for rtdetr-0.5.0-py3-none-any.whl
Algorithm Hash digest
SHA256 68df257e4918f6342a0e844bf780b8f1f4b2840b2158eaaf01ad4f3f7d4bc810
MD5 1acc4d6c1a2d88ab640245178dfede4d
BLAKE2b-256 aa3c26f2417f76ecbcee9441c40bcaa15b55b2be2de9bbf99b754ff9ed217ce7

See more details on using hashes here.

Provenance

The following attestation bundles were made for rtdetr-0.5.0-py3-none-any.whl:

Publisher: publish.yml on leeyunjai82/rtdetr

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

0.6.2

2 files

0.6.1

2 files

0.6.0

2 files

This release

0.5.0 This release

2 files

0.4.0

2 files

0.3.2

2 files

0.3.1

2 files

0.3.0

2 files

0.2.1

2 files

0.2.0

2 files

0.1.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page