Adding TransUNet Code and Model
Showing
- TransUNet/README.md 49 additions, 0 deletionsTransUNet/README.md
- TransUNet/Training TransUNet on US data.ipynb 11526 additions, 0 deletionsTransUNet/Training TransUNet on US data.ipynb
- TransUNet/TransUNet/LICENSE 201 additions, 0 deletionsTransUNet/TransUNet/LICENSE
- TransUNet/TransUNet/datasets/dataset_sat.py 71 additions, 0 deletionsTransUNet/TransUNet/datasets/dataset_sat.py
- TransUNet/TransUNet/lists/lists_SatImgs/train.txt 3072 additions, 0 deletionsTransUNet/TransUNet/lists/lists_SatImgs/train.txt
- TransUNet/TransUNet/lists/lists_SatImgs/val.txt 257 additions, 0 deletionsTransUNet/TransUNet/lists/lists_SatImgs/val.txt
- TransUNet/TransUNet/lists/lists_kielstau/train.txt 5106 additions, 0 deletionsTransUNet/TransUNet/lists/lists_kielstau/train.txt
- TransUNet/TransUNet/loss.py 310 additions, 0 deletionsTransUNet/TransUNet/loss.py
- TransUNet/TransUNet/networks/vit_seg_configs.py 130 additions, 0 deletionsTransUNet/TransUNet/networks/vit_seg_configs.py
- TransUNet/TransUNet/networks/vit_seg_modeling.py 453 additions, 0 deletionsTransUNet/TransUNet/networks/vit_seg_modeling.py
- TransUNet/TransUNet/networks/vit_seg_modeling_resnet_skip.py 160 additions, 0 deletionsTransUNet/TransUNet/networks/vit_seg_modeling_resnet_skip.py
- TransUNet/TransUNet/requirements.txt 12 additions, 0 deletionsTransUNet/TransUNet/requirements.txt
- TransUNet/TransUNet/train_sat.py 102 additions, 0 deletionsTransUNet/TransUNet/train_sat.py
- TransUNet/TransUNet/trainer_sat.py 219 additions, 0 deletionsTransUNet/TransUNet/trainer_sat.py
- TransUNet/TransUNet/utils.py 80 additions, 0 deletionsTransUNet/TransUNet/utils.py
- TransUNet/weights/epoch_150.pth 0 additions, 0 deletionsTransUNet/weights/epoch_150.pth
Loading
Please register or sign in to comment