├── .idea ├── C2FTrans.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml ├── vcs.xml └── workspace.xml ├── README.md ├── models ├── MPtrans.py ├── __init__.py ├── transformer_parts.py ├── transformer_parts_mp.py └── unets_parts.py ├── requirements.txt ├── test.py ├── train.py └── utils ├── flops_counter.py ├── imgname.py ├── loss_functions ├── TopK_loss.py ├── crossentropy.py └── dice_loss.py ├── metrics.py ├── nd_softmax.py ├── non_maximum_suppression.py ├── tensor_utils.py ├── transfer_model.py ├── utils_gray.py ├── utils_multi.py ├── utils_multi_rgb.py ├── utils_rgb.py └── visualization.py /.idea/C2FTrans.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/.idea/C2FTrans.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /.idea/workspace.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/.idea/workspace.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/README.md -------------------------------------------------------------------------------- /models/MPtrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/models/MPtrans.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .MPtrans import * -------------------------------------------------------------------------------- /models/transformer_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/models/transformer_parts.py -------------------------------------------------------------------------------- /models/transformer_parts_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/models/transformer_parts_mp.py -------------------------------------------------------------------------------- /models/unets_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/models/unets_parts.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/train.py -------------------------------------------------------------------------------- /utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/flops_counter.py -------------------------------------------------------------------------------- /utils/imgname.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/imgname.py -------------------------------------------------------------------------------- /utils/loss_functions/TopK_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/loss_functions/TopK_loss.py -------------------------------------------------------------------------------- /utils/loss_functions/crossentropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/loss_functions/crossentropy.py -------------------------------------------------------------------------------- /utils/loss_functions/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/loss_functions/dice_loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/nd_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/nd_softmax.py -------------------------------------------------------------------------------- /utils/non_maximum_suppression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/non_maximum_suppression.py -------------------------------------------------------------------------------- /utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/tensor_utils.py -------------------------------------------------------------------------------- /utils/transfer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/transfer_model.py -------------------------------------------------------------------------------- /utils/utils_gray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/utils_gray.py -------------------------------------------------------------------------------- /utils/utils_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/utils_multi.py -------------------------------------------------------------------------------- /utils/utils_multi_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/utils_multi_rgb.py -------------------------------------------------------------------------------- /utils/utils_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/utils_rgb.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xianlin7/BATFormer/HEAD/utils/visualization.py --------------------------------------------------------------------------------