transformers/examples/pytorch/semantic-segmentation/requirements.txt
Pavel Iakubovskii 56d001b26f
Fix and simplify semantic-segmentation example (#30145)
* Remove unused augmentation

* Fix pad_if_smaller() and remove unused augmentation

* Add indentation

* Fix requirements

* Update dataset use instructions

* Replace transforms with albumentations

* Replace identity transform with None

* Fixing formatting

* Fixed comment place
2024-04-10 09:10:52 +01:00

6 lines
72 B
Plaintext

datasets >= 2.0.0
torch >= 1.3
accelerate
evaluate
Pillow
albumentations