GenSeg-Baselines / code /envs /seggen-controlnet.yml
MaybeRichard's picture
code: complete eval pipeline (7 metrics + per-class + Wilcoxon) + Swin-UNet/TransUNet networks; remove backups/obsolete
1a18f22 verified
Raw
History Blame Contribute Delete
656 Bytes
# Isolated env for ControlNet (pytorch-lightning 1.5 + old torch). Needs SD v1.5 ckpt.
# Generative-augmentation baseline only; never mix with the main env.
name: seggen-controlnet
channels: [conda-forge]
dependencies:
- python=3.10
- pip
- pip:
# torch 1.12.1 (cu113 wheels bundle their own runtime):
# pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
- pytorch-lightning==1.5.0
- omegaconf==2.1.1
- einops==0.3.0
- transformers==4.19.2
- open-clip-torch==2.0.2
- gradio==3.16.2
- opencv-python-headless
- basicsr==1.4.2