Segmentation support & other enchancements (#40)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ayush Chaurasia 2022-11-08 20:57:57 +05:30 committed by GitHub
parent c617ee1c79
commit f56c9bcc26
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
17 changed files with 1320 additions and 47 deletions

View file

@ -1,7 +1,7 @@
from pathlib import Path
from ultralytics.yolo.v8 import classify
from ultralytics.yolo.v8 import classify, segment
ROOT = Path(__file__).parents[0] # yolov8 ROOT
__all__ = ["classify"]
__all__ = ["classify", "segment"]