Celltool_public / requirements.txt
Shengxiao0709's picture
Update requirements.txt
cfcb7a7 verified
Raw History Blame Contribute Delete
1.01 kB
# PyTorch 2.4.1 + torchvision
torch==2.4.1
torchvision==0.19.1
torchaudio==2.4.1
# Core dependencies
diffusers
transformers>=4.34.0
accelerate
pyrallis
easydict
omegaconf==2.1.1
einops==0.3.0
torch-fidelity==0.3.0
torchmetrics>=0.11.0
pytorch-lightning==2.0.9
taming-transformers @ git+https://github.com/CompVis/taming-transformers.git@master
clip @ git+https://github.com/openai/CLIP.git@main
# Computer vision
opencv-python==4.5.5.64 # ✅ 避免 dnn.DictValue 错误
opencv-python-headless==4.5.5.64 # ✅ 防止系统默认装新版导致冲突(和 GUI 冲突不大)
kornia==0.6
albumentations==0.4.3
imageio>=2.27 # ✅ 保证兼容 scikit-image 0.21.0
imageio-ffmpeg==0.4.2
matplotlib
scikit-image==0.21.0
Pillow
segment-anything
numpy==1.24.4 # ✅ 避免 numpy 2.x 引发 ABI 问题
# Gradio
gradio
gradio-bbox-annotator
# Utilities
natsort
roifile
fill-voids
configargparse
ipywidgets
ftfy
sniffio
websocket-client
dask