WSCL / requirements.txt
yhzhai's picture
append probablities to the output
2f7bb6a
raw
history blame
560 Bytes
albumentations==1.0.0
einops==0.4.1
fast_pytorch_kmeans==0.1.6
glob2==0.7
gpustat==0.6.0
h5py==3.6.0
matplotlib==3.3.4
numpy==1.22.4
opencv_contrib_python==4.5.3.56
opencv_python==4.4.0.46
opencv_python_headless==4.5.3.56
pandas==1.3.5
Pillow==9.4.0
prettytable==2.2.1
pydensecrf==1.0rc2
PyYAML==5.4.1
scikit_image==0.18.3
scikit_learn==0.24.1
scipy==1.7.3
spatial_correlation_sampler==0.4.0
SQLAlchemy==1.4.15
sync_batchnorm==0.0.1
tensorboard==2.12.2
termcolor==2.4.0
timm==0.9.12
torch==1.12.1+cu116
torchvision==0.13.1+cu116
tqdm==4.64.1
markupsafe==2.0.1