Skip to content

Commit

Permalink
Config specs
Browse files Browse the repository at this point in the history
  • Loading branch information
Zachary Adler committed May 30, 2024
1 parent b4f8c62 commit 148462a
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions sahi/models/yolov8engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,20 @@
from sahi.utils.compatibility import fix_full_shape_list, fix_shift_amount_list
from sahi.utils.import_utils import check_requirements

# Must be passed a config_path to a json with the below fields.
# Fields must be obtained from base PyTorch model before quantization.
# {
# "task": "segment", # or 'detect'
# "names": [
# "dog",
# "cat",
# ],
# "imgsz": [
# 640,
# 640
# ],
# "half": true
# }

class Yolov8EngineDetectionModel(DetectionModel):
def __init__(self):
Expand Down

0 comments on commit 148462a

Please sign in to comment.