RTMDet Pytorch Implementation
Project description
RTMDet – PyTorch Implementation
This repository is a PyTorch port of RTMDet, originally implemented in MMDetection.
The goal is to reimplement the network in pure PyTorch while making it possible to load pretrained weights from the original models.
Installation
pip install rtmdet
Usage
from rtmdet import RTMDet
model = RTMDet.from_preset("small") # tiny / small / medium / large
bboxes, scores, classes = model("image.jpg")
References
- RTMDet: An Empirical Study of Real-Time Object Detectors
Xiangyu Zhang, Xinyu Zhou, Zhiqi Li, et al.
📄 Paper
Acknowledgments
Based on MMDetection and the official RTMDet implementation.
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
rtmdet-0.2.1.tar.gz
(14.1 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
rtmdet-0.2.1-py3-none-any.whl
(18.9 kB
view details)
File details
Details for the file rtmdet-0.2.1.tar.gz.
File metadata
- Download URL: rtmdet-0.2.1.tar.gz
- Upload date:
- Size: 14.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.8.24
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8153c2b4926a838139ea67b7915025aba0073aef9d90c8403b62d7972cf786eb
|
|
| MD5 |
37c2dae39a5f09a7de2a8bcd961b27ed
|
|
| BLAKE2b-256 |
be763d1160aa110f7a7811115eb0daed32bce3afff55181ce7fdae7302f854f2
|
File details
Details for the file rtmdet-0.2.1-py3-none-any.whl.
File metadata
- Download URL: rtmdet-0.2.1-py3-none-any.whl
- Upload date:
- Size: 18.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.8.24
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8772525e377329d68c9f607e069b17cebc3f2db1e86ddb614e66976dc87e95e7
|
|
| MD5 |
76bace960db7692fc9d4de3b508636df
|
|
| BLAKE2b-256 |
60500b6decd0ea03396c9862b06407400125a071a038ddc103c83e8794b6eda9
|
