kadirnar commited on
Commit
04119c7
·
1 Parent(s): a15113c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -9
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/yolov6-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,25 +22,20 @@ datasets:
22
  pip install yoloxdetect
23
  ```
24
 
25
- ### Yolov6 Inference
26
  ```python
27
- from yoloxdetect import YoloxDetect
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