浏览代码

Common `is_coco` logic betwen train.py and val.py (#5521)

modifyDataloader
Glenn Jocher GitHub 3 年前
父节点
当前提交
bcc085d83f
找不到此签名对应的密钥 GPG 密钥 ID: 4AEE18F83AFDEB23
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. +1
    -1
      train.py

+ 1
- 1
train.py 查看文件

@@ -107,7 +107,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
nc = 1 if single_cls else int(data_dict['nc']) # number of classes
names = ['item'] if single_cls and len(data_dict['names']) != 1 else data_dict['names'] # class names
assert len(names) == nc, f'{len(names)} names found for nc={nc} dataset in {data}' # check
is_coco = data.endswith('coco.yaml') and nc == 80 # COCO dataset
is_coco = isinstance(val_path, str) and val_path.endswith('coco/val2017.txt') # COCO dataset

# Model
check_suffix(weights, '.pt') # check weights

正在加载...
取消
保存