Update README.md
Browse files
README.md
CHANGED
|
@@ -13,7 +13,7 @@ datasets:
|
|
| 13 |
### Model Description
|
| 14 |
[YOLOX](https://arxiv.org/abs/2107.08430) is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported.
|
| 15 |
|
| 16 |
-
[YOLOXDetect-Pip](https://github.com/kadirnar/
|
| 17 |
|
| 18 |
[Paper Repo]: Implementation of paper - [YOLOX](https://github.com/Megvii-BaseDetection/YOLOX)
|
| 19 |
|
|
@@ -22,25 +22,20 @@ datasets:
|
|
| 22 |
pip install yoloxdetect
|
| 23 |
```
|
| 24 |
|
| 25 |
-
###
|
| 26 |
```python
|
| 27 |
-
from yoloxdetect import
|
| 28 |
from yolox.data.datasets import COCO_CLASSES
|
| 29 |
-
|
| 30 |
-
model = YoloxDetect(
|
| 31 |
model_path = "kadirnar/yolox_s-v0.1.1",
|
| 32 |
config_path = "configs.yolox_s",
|
| 33 |
device = "cuda:0",
|
| 34 |
-
classes = COCO_CLASSES,
|
| 35 |
-
confidence_threshold = 0.25,
|
| 36 |
-
nms_threshold = 0.45,
|
| 37 |
)
|
| 38 |
model.classes = COCO_CLASSES
|
| 39 |
model.conf = 0.25
|
| 40 |
model.iou = 0.45
|
| 41 |
model.show = False
|
| 42 |
model.save = True
|
| 43 |
-
|
| 44 |
pred = model.predict(image='data/images', img_size=640)
|
| 45 |
```
|
| 46 |
|
|
|
|
| 13 |
### Model Description
|
| 14 |
[YOLOX](https://arxiv.org/abs/2107.08430) is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported.
|
| 15 |
|
| 16 |
+
[YOLOXDetect-Pip](https://github.com/kadirnar/yolox-pip/): This repo is a packaged version of the [YOLOX](https://github.com/Megvii-BaseDetection/YOLOX) for easy installation and use.
|
| 17 |
|
| 18 |
[Paper Repo]: Implementation of paper - [YOLOX](https://github.com/Megvii-BaseDetection/YOLOX)
|
| 19 |
|
|
|
|
| 22 |
pip install yoloxdetect
|
| 23 |
```
|
| 24 |
|
| 25 |
+
### Yolox Inference
|
| 26 |
```python
|
| 27 |
+
from yoloxdetect import YoloxDetector
|
| 28 |
from yolox.data.datasets import COCO_CLASSES
|
| 29 |
+
model = YoloxDetector(
|
|
|
|
| 30 |
model_path = "kadirnar/yolox_s-v0.1.1",
|
| 31 |
config_path = "configs.yolox_s",
|
| 32 |
device = "cuda:0",
|
|
|
|
|
|
|
|
|
|
| 33 |
)
|
| 34 |
model.classes = COCO_CLASSES
|
| 35 |
model.conf = 0.25
|
| 36 |
model.iou = 0.45
|
| 37 |
model.show = False
|
| 38 |
model.save = True
|
|
|
|
| 39 |
pred = model.predict(image='data/images', img_size=640)
|
| 40 |
```
|
| 41 |
|