├── README.md ├── cal_IS.py ├── cmd.py ├── dataset.py ├── evaluate.py ├── models ├── DiffAugment.py ├── Discriminator_ml.py ├── build4.py ├── layer ├── selfattention1.py └── swin_transformer4.py ├── train_transformer4.py └── utils ├── loss.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/README.md -------------------------------------------------------------------------------- /cal_IS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/cal_IS.py -------------------------------------------------------------------------------- /cmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/cmd.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/dataset.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/evaluate.py -------------------------------------------------------------------------------- /models/DiffAugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/models/DiffAugment.py -------------------------------------------------------------------------------- /models/Discriminator_ml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/models/Discriminator_ml.py -------------------------------------------------------------------------------- /models/build4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/models/build4.py -------------------------------------------------------------------------------- /models/layer: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/models/layer -------------------------------------------------------------------------------- /models/selfattention1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/models/selfattention1.py -------------------------------------------------------------------------------- /models/swin_transformer4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/models/swin_transformer4.py -------------------------------------------------------------------------------- /train_transformer4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/train_transformer4.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PengleiGao/UTransformer/HEAD/utils/utils.py --------------------------------------------------------------------------------