YOLOv5 + Albumentations integration (#3882)

* Albumentations integration

* ToGray p=0.01

* print confirmation

* create instance in dataloader init method

* improved version handling

* transform not defined fix

* assert string update

* create check_version()

* add spaces

* update class comment
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-05 18:01:54 +02:00
1 parent 6a3ee7cf03
commit 33202b7f0b
4 files changed
+61 -27

No files matched your search

+1
View File
@@ -27,4 +27,5 @@ pandas
# extras --------------------------------------
# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
# pycocotools>=2.0 # COCO mAP
# albumentations>=1.0.0
thop # FLOPs computation