ST-UNet

Pytorch codes for 'Swin Transformer Embedding UNet for Remote Sensing Image Semantic Segmentation'

Environment

conda create -n xxx python=3.7 -y

conda activate xxx

pip install -r requirements.txt

requirements.txt

torchviz
torchvision==0.5.0
torch==1.4.0
timm==0.3.2
numpy
tqdm
tensorboard
tensorboardX
ml-collections
medpy
SimpleITK
scipy
h5py
opencv-python
pillow
toposort
tifffile
matplotlib
yacs 
einops

Reference

• TransUNet
• SwinTransformer
• SoftPool