├── README.md ├── data ├── data_aug.py ├── test_select.py └── trainval_split.py ├── dataset.py ├── imgs ├── att_ds.jpg ├── attunet.jpg ├── label_0.png ├── label_12.png ├── label_4.png └── models.png ├── logs └── Attention_Unet_Vgg │ ├── loss_2021_09_20_08_45_53.png │ ├── preout │ └── results.txt │ ├── train_info_2021_09_20_08_45_53.txt │ └── val_info_2021_09_20_08_45_53.txt ├── model ├── Attention_Unet_Vgg.py ├── Attention_ds_Unet_Vgg.py ├── Base_Unet_Vgg_Official.py ├── Base_Unet_Vgg_Simple1.py ├── Base_Unet_Vgg_Simple2.py ├── __init__.py ├── arch_plot │ ├── Base_Unet_Vgg_Official │ │ └── events.out.tfevents.1630843112.DESKTOP-4E95S3N.20528.0 │ └── Base_Unet_Vgg_Simple │ │ └── events.out.tfevents.1631007565.DESKTOP-4E95S3N.12268.0 └── backbone │ └── vgg16.py ├── predict.py ├── predict_ds.py ├── reference └── readme.md ├── train.py ├── train_ds.py └── utils ├── __init__.py ├── logger.py ├── loss.py └── metrics.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/README.md -------------------------------------------------------------------------------- /data/data_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/data/data_aug.py -------------------------------------------------------------------------------- /data/test_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/data/test_select.py -------------------------------------------------------------------------------- /data/trainval_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/data/trainval_split.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/dataset.py -------------------------------------------------------------------------------- /imgs/att_ds.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/imgs/att_ds.jpg -------------------------------------------------------------------------------- /imgs/attunet.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/imgs/attunet.jpg -------------------------------------------------------------------------------- /imgs/label_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/imgs/label_0.png -------------------------------------------------------------------------------- /imgs/label_12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/imgs/label_12.png -------------------------------------------------------------------------------- /imgs/label_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/imgs/label_4.png -------------------------------------------------------------------------------- /imgs/models.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/imgs/models.png -------------------------------------------------------------------------------- /logs/Attention_Unet_Vgg/loss_2021_09_20_08_45_53.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/logs/Attention_Unet_Vgg/loss_2021_09_20_08_45_53.png -------------------------------------------------------------------------------- /logs/Attention_Unet_Vgg/preout/results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/logs/Attention_Unet_Vgg/preout/results.txt -------------------------------------------------------------------------------- /logs/Attention_Unet_Vgg/train_info_2021_09_20_08_45_53.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/logs/Attention_Unet_Vgg/train_info_2021_09_20_08_45_53.txt -------------------------------------------------------------------------------- /logs/Attention_Unet_Vgg/val_info_2021_09_20_08_45_53.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/logs/Attention_Unet_Vgg/val_info_2021_09_20_08_45_53.txt -------------------------------------------------------------------------------- /model/Attention_Unet_Vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/Attention_Unet_Vgg.py -------------------------------------------------------------------------------- /model/Attention_ds_Unet_Vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/Attention_ds_Unet_Vgg.py -------------------------------------------------------------------------------- /model/Base_Unet_Vgg_Official.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/Base_Unet_Vgg_Official.py -------------------------------------------------------------------------------- /model/Base_Unet_Vgg_Simple1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/Base_Unet_Vgg_Simple1.py -------------------------------------------------------------------------------- /model/Base_Unet_Vgg_Simple2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/Base_Unet_Vgg_Simple2.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/arch_plot/Base_Unet_Vgg_Official/events.out.tfevents.1630843112.DESKTOP-4E95S3N.20528.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/arch_plot/Base_Unet_Vgg_Official/events.out.tfevents.1630843112.DESKTOP-4E95S3N.20528.0 -------------------------------------------------------------------------------- /model/arch_plot/Base_Unet_Vgg_Simple/events.out.tfevents.1631007565.DESKTOP-4E95S3N.12268.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/arch_plot/Base_Unet_Vgg_Simple/events.out.tfevents.1631007565.DESKTOP-4E95S3N.12268.0 -------------------------------------------------------------------------------- /model/backbone/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/model/backbone/vgg16.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/predict.py -------------------------------------------------------------------------------- /predict_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/predict_ds.py -------------------------------------------------------------------------------- /reference/readme.md: -------------------------------------------------------------------------------- 1 | # 这里是模型搭建过程中参考的文献 2 | 3 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/train.py -------------------------------------------------------------------------------- /train_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/train_ds.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sucaicai4/Unet-series-for-Ultrasound-image-segmentation/HEAD/utils/metrics.py --------------------------------------------------------------------------------