Wnet / requirements.txt
dongjoaquin's picture
Upload requirements.txt with huggingface_hub
0b65ac5 verified
# RadioUNet SoundMapDiff训练依赖
# 核心深度学习框架
torch>=1.12.0
torchvision>=0.13.0
# 数据处理
numpy>=1.21.0
pandas>=1.3.0
pillow>=9.0.0
# 可视化
matplotlib>=3.5.0
seaborn>=0.11.0
# 工具库
tqdm>=4.62.0
scikit-image>=0.19.0
opencv-python>=4.5.0
# 其他
pyyaml>=5.4.0