Skip to main content

It is a demo application of the YOLOX model.

Project description

Yolox-Pip: This is a packaged version of the YOLOX for easy installation and use.

teaser

Overview

This repo is a packaged version of the YOLOX for easy installation and use.

Installation

pip install yoloxdetect

Yolox Inference

from yoloxdetect import YoloxDetector

model = YoloxDetector(
    model_path = "kadirnar/yolox_s-v0.1.1", # or "data/weights/yolox_s.pth"
    config_path = "configs.yolox_s",
    device = "cuda:0",
    hf_model=True,
)
model.classes = None
model.conf = 0.25
model.iou = 0.45
model.show = False
model.save = True

pred = model.predict(image='data/images', img_size=640)

Citation

 @article{yolox2021,
  title={YOLOX: Exceeding YOLO Series in 2021},
  author={Ge, Zheng and Liu, Songtao and Wang, Feng and Li, Zeming and Sun, Jian},
  journal={arXiv preprint arXiv:2107.08430},
  year={2021}
}

Project details


Download files

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

Source Distribution

yoloxdetect-0.0.10.tar.gz (7.8 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page