Sfoglia il codice sorgente

`int(mlc)` (#4385)

modifyDataloader
Glenn Jocher GitHub 3 anni fa
parent
commit
ce7deec440
Non sono state trovate chiavi note per questa firma nel database ID Chiave GPG: 4AEE18F83AFDEB23
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. +1
    -1
      train.py

+ 1
- 1
train.py Vedi File

@@ -204,7 +204,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
hyp=hyp, augment=True, cache=opt.cache, rect=opt.rect, rank=RANK,
workers=workers, image_weights=opt.image_weights, quad=opt.quad,
prefix=colorstr('train: '))
mlc = np.concatenate(dataset.labels, 0)[:, 0].max() # max label class
mlc = int(np.concatenate(dataset.labels, 0)[:, 0].max()) # max label class
nb = len(train_loader) # number of batches
assert mlc < nc, f'Label class {mlc} exceeds nc={nc} in {data}. Possible class labels are 0-{nc - 1}'


Loading…
Annulla
Salva