xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

YOLOv5 updates (#90)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2022-12-25 14:33:18 +01:00
committed by GitHub
parent ebd3cfb2fd
commit 98815d560f
27 changed files with 281 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/tests/data/dataloader/yolodetection.py
View File

@ -55,7 +55,7 @@ def plot_one_box(x, img, color=None, label=None, line_thickness=None):
)
@hydra.main(version_base=None, config_path=DEFAULT_CONFIG.parent, config_name=DEFAULT_CONFIG.name)
@hydra.main(version_base=None, config_path=str(DEFAULT_CONFIG.parent), config_name=DEFAULT_CONFIG.name)
def test(cfg):
cfg.task = "detect"
cfg.mode = "train"

2
ultralytics/tests/data/dataloader/yolosegment.py
View File

@ -54,7 +54,7 @@ def plot_one_box(x, img, color=None, label=None, line_thickness=None):
)
@hydra.main(version_base=None, config_path=DEFAULT_CONFIG.parent, config_name=DEFAULT_CONFIG.name)
@hydra.main(version_base=None, config_path=str(DEFAULT_CONFIG.parent), config_name=DEFAULT_CONFIG.name)
def test(cfg):
cfg.task = "segment"
cfg.mode = "train"