ultralytics 8.0.30 Docker, rect, data=*.zip updates (#832)
Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
This commit is contained in:
parent
09265b17d7
commit
64f247d692
17 changed files with 67 additions and 43 deletions
|
|
@ -22,7 +22,6 @@ class DetectionValidator(BaseValidator):
|
|||
def __init__(self, dataloader=None, save_dir=None, pbar=None, logger=None, args=None):
|
||||
super().__init__(dataloader, save_dir, pbar, logger, args)
|
||||
self.args.task = 'detect'
|
||||
self.data_dict = yaml_load(check_file(self.args.data), append_filename=True) if self.args.data else None
|
||||
self.is_coco = False
|
||||
self.class_map = None
|
||||
self.metrics = DetMetrics(save_dir=self.save_dir)
|
||||
|
|
@ -172,7 +171,7 @@ class DetectionValidator(BaseValidator):
|
|||
hyp=vars(self.args),
|
||||
cache=False,
|
||||
pad=0.5,
|
||||
rect=True,
|
||||
rect=self.args.rect,
|
||||
workers=self.args.workers,
|
||||
prefix=colorstr(f'{self.args.mode}: '),
|
||||
shuffle=False,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue