Metadata-Version: 2.1
Name: det_executor
Version: 0.0.2
Summary: Python package with latest versions of YOLO architecture for training and inference
Home-page: https://github.com/maxwolf8852/DetExecutor.git
Author: Maxim Volkovskiy
Author-email: maxwolf8852@gmail.com
Classifier: Programming Language :: Python :: 3.6
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown
License-File: LICENSE

# DetExecutor
Python package with latest versions of YOLO architecture for training and inference
## Install
Installing is quite simple, just use pip:
```shell
pip3 install det_executor
```
## Train
Training support is still in progress!
## Inference
### Loading model
```python3
from det_executor import DetExecutor
# print list of supported arches
DetExecutor.list_arch()

# loading model
name = 'yolov7'
ex = DetExecutor(name)
```
### Predict and draw
```python3
from det_executor import DetExecutor, draw_on_image
import cv2

# loading model
name = 'yolov7'
ex = DetExecutor(name)

# loading image
img = cv2.imread('test/img.jpg')

# predict
classes, boxes, scores = ex.predict(img)

# draw
img = draw_on_image(img, boxes[0], scores[0], classes[0])
cv2.imshow("image", img)
cv2.waitKey()
```
