PyTorch implementation of YOLOv5, YOLOv6, YOLOv7, YOLOX
Project description
Introduction
The TorchYolo library aims to support all YOLO models(like YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOX, etc.) and provide a unified interface for training and inference. The library is based on PyTorch and is designed to be easy to use and extend.
Installation
pip install torchyolo
Prediction
from torchyolo import YoloHub
predictor = YoloHub(
model_type="yolov5",
model_path="yolov5s.pt",
device='cpu',
image_size=640
)
predictor.conf_thres = 0.25
predictor.iou_thres = 0.45
predictor.save = True
predictor.show = False
image = "data/highway.jpg"
result = predictor.predict(image)
Model Architecture
from torchyolo import YoloHub
model = YoloHub(
model_type="yolov8",
model_path="yolov8n.pt",
device="cuda:0",
image_size=640)
result = model.view_model(file_format="pdf")
Contributing
Before opening a PR:
- Install required development packages:
pip install -r requirements-dev.txt
- Reformat the code with black and isort:
bash script/code_format.sh
Acknowledgement
A part of the code is borrowed from SAHI. Many thanks for their wonderful works.
Citation
@article{li2022yolov6,
title={YOLOv6: A single-stage object detection framework for industrial applications},
author={Li, Chuyi and Li, Lulu and Jiang, Hongliang and Weng, Kaiheng and Geng, Yifei and Li, Liang and Ke, Zaidan and Li, Qingyuan and Cheng, Meng and Nie, Weiqiang and others},
journal={arXiv preprint arXiv:2209.02976},
year={2022}
}
@article{wang2022yolov7,
title={{YOLOv7}: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors},
author={Wang, Chien-Yao and Bochkovskiy, Alexey and Liao, Hong-Yuan Mark},
journal={arXiv preprint arXiv:2207.02696},
year={2022}
}
@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}
}
@software{glenn_jocher_2020_4154370,
author = {Glenn Jocher and,Alex Stoken and,Jirka Borovec and,NanoCode012 and,ChristopherSTAN and,Liu Changyu and,Laughing and,tkianai and,Adam Hogan and,lorenzomammana and,yxNONG and,AlexWang1900 and,Laurentiu Diaconu and,Marc and,wanghaoyang0106 and,ml5ah and,Doug and,Francisco Ingham and,Frederik and,Guilhen and,Hatovix and,Jake Poznanski and,Jiacong Fang and,Lijun Yu 于力军 and,changyu98 and,Mingyu Wang and,Naman Gupta and,Osama Akhtar and,PetrDvoracek and,Prashant Rai},
title={{ultralytics/yolov5: v7.2 - Bug Fixes and
Performance Improvements}},
month= oct,
year= 2020,
publisher= {Zenodo},
version= {v3.1},
doi= {10.5281/zenodo.4154370},
url= {https://doi.org/10.5281/zenodo.4154370}
}
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchyolo-0.2.6.tar.gz
(24.6 kB
view hashes)
