init

finetune/requirements/albu.txt

@@ -0,0 +1 @@
+albumentations>=0.3.2 --no-binary qudida,albumentations

finetune/requirements/docs.txt

@@ -0,0 +1,7 @@
+docutils==0.16.0
+myst-parser
+-e git+https://github.com/open-mmlab/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
+sphinx==4.0.2
+sphinx_copybutton
+sphinx_markdown_tables
+urllib3<2.0.0

finetune/requirements/mminstall.txt

@@ -0,0 +1,2 @@
+mmcv>=2.0.0rc4,<2.2.0
+mmengine>=0.5.0,<1.0.0

finetune/requirements/multimodal.txt

@@ -0,0 +1,2 @@
+ftfy
+regex

finetune/requirements/optional.txt

@@ -0,0 +1,22 @@
+cityscapesscripts
+-e git+https://github.com/openai/CLIP.git@main#egg=clip
+
+# for vpd model
+diffusers
+einops==0.3.0
+imageio==2.9.0
+imageio-ffmpeg==0.4.2
+invisible-watermark
+kornia==0.6
+-e git+https://github.com/CompVis/stable-diffusion@21f890f#egg=latent-diffusion
+nibabel
+omegaconf==2.1.1
+pudb==2019.2
+pytorch-lightning==1.4.2
+streamlit>=0.73.1
+-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
+test-tube>=0.7.5
+timm
+torch-fidelity==0.3.0
+torchmetrics==0.6.0
+transformers==4.19.2

finetune/requirements/readthedocs.txt

@@ -0,0 +1,6 @@
+mmcv>=2.0.0rc1,<2.1.0
+mmengine>=0.4.0,<1.0.0
+prettytable
+scipy
+torch
+torchvision

finetune/requirements/runtime.txt

@@ -0,0 +1,5 @@
+matplotlib
+numpy
+packaging
+prettytable
+scipy

finetune/requirements/tests.txt

@@ -0,0 +1,8 @@
+codecov
+flake8
+ftfy
+interrogate
+pytest
+regex
+xdoctest>=0.10.0
+yapf