├── README.md ├── config.py ├── configs └── swin_tiny_patch4_window7_224_lite.yaml ├── data_demo ├── classification │ └── UDIAT │ │ ├── 0 │ │ └── demo_01.png │ │ ├── 1 │ │ └── demo_02.png │ │ ├── config.yaml │ │ └── test.txt └── segmentation │ └── BUSIS │ ├── config.yaml │ ├── imgs │ ├── demo_01.png │ └── demo_02.png │ ├── masks │ ├── demo_01.png │ └── demo_02.png │ └── test.txt ├── datasets ├── dataset.py └── omni_dataset.py ├── networks └── omni_vision_transformer.py ├── omni_test.py ├── omni_train.py ├── omni_trainer.py ├── pretrained_ckpt └── .gitkeeep ├── requirements.txt └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/config.py -------------------------------------------------------------------------------- /configs/swin_tiny_patch4_window7_224_lite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/configs/swin_tiny_patch4_window7_224_lite.yaml -------------------------------------------------------------------------------- /data_demo/classification/UDIAT/0/demo_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/classification/UDIAT/0/demo_01.png -------------------------------------------------------------------------------- /data_demo/classification/UDIAT/1/demo_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/classification/UDIAT/1/demo_02.png -------------------------------------------------------------------------------- /data_demo/classification/UDIAT/config.yaml: -------------------------------------------------------------------------------- 1 | 0:Benign 2 | 1:Malignant 3 | -------------------------------------------------------------------------------- /data_demo/classification/UDIAT/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/classification/UDIAT/test.txt -------------------------------------------------------------------------------- /data_demo/segmentation/BUSIS/config.yaml: -------------------------------------------------------------------------------- 1 | 0:background:0 2 | 1:nodule:255 3 | -------------------------------------------------------------------------------- /data_demo/segmentation/BUSIS/imgs/demo_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/segmentation/BUSIS/imgs/demo_01.png -------------------------------------------------------------------------------- /data_demo/segmentation/BUSIS/imgs/demo_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/segmentation/BUSIS/imgs/demo_02.png -------------------------------------------------------------------------------- /data_demo/segmentation/BUSIS/masks/demo_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/segmentation/BUSIS/masks/demo_01.png -------------------------------------------------------------------------------- /data_demo/segmentation/BUSIS/masks/demo_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/segmentation/BUSIS/masks/demo_02.png -------------------------------------------------------------------------------- /data_demo/segmentation/BUSIS/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/data_demo/segmentation/BUSIS/test.txt -------------------------------------------------------------------------------- /datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/datasets/dataset.py -------------------------------------------------------------------------------- /datasets/omni_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/datasets/omni_dataset.py -------------------------------------------------------------------------------- /networks/omni_vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/networks/omni_vision_transformer.py -------------------------------------------------------------------------------- /omni_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/omni_test.py -------------------------------------------------------------------------------- /omni_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/omni_train.py -------------------------------------------------------------------------------- /omni_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/omni_trainer.py -------------------------------------------------------------------------------- /pretrained_ckpt/.gitkeeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zehui-Lin/UniUSNet/HEAD/utils.py --------------------------------------------------------------------------------