Pārlūkot izejas kodu

Update augmentations.py (#3948)

modifyDataloader
Glenn Jocher GitHub pirms 3 gadiem
vecāks
revīzija
a26e7de2bf
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. +1
    -1
      utils/augmentations.py

+ 1
- 1
utils/augmentations.py Parādīt failu

@@ -25,7 +25,7 @@ class Albumentations:
A.ToGray(p=0.01)],
bbox_params=A.BboxParams(format='yolo', label_fields=['class_labels']))

logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms))
logging.info(colorstr('albumentations: ') + ', '.join(f'{x}' for x in self.transform.transforms if x.p))
except ImportError: # package not installed, skip
pass
except Exception as e:

Notiek ielāde…
Atcelt
Saglabāt