├── .gitignore ├── Data ├── test_samples │ ├── img │ │ ├── austin2_01179_258_289.png │ │ ├── austin3_00294_332_293.png │ │ ├── austin3_00497_432_361.png │ │ ├── austin3_01452_334_335.png │ │ ├── austin4_00061_481_276.png │ │ ├── austin4_00133_264_308.png │ │ ├── austin4_00163_356_367.png │ │ ├── austin4_00205_468_478.png │ │ ├── austin7_00079_495_463.png │ │ ├── austin7_00100_300_263.png │ │ └── austin7_00166_430_319.png │ └── label │ │ ├── austin2_01179_258_289.png │ │ ├── austin3_00294_332_293.png │ │ ├── austin3_00497_432_361.png │ │ ├── austin3_01452_334_335.png │ │ ├── austin4_00061_481_276.png │ │ ├── austin4_00133_264_308.png │ │ ├── austin4_00163_356_367.png │ │ ├── austin4_00205_468_478.png │ │ ├── austin7_00079_495_463.png │ │ ├── austin7_00100_300_263.png │ │ └── austin7_00166_430_319.png ├── train_samples │ ├── img │ │ ├── austin2_01179_258_289.png │ │ ├── austin3_00294_332_293.png │ │ ├── austin3_00497_432_361.png │ │ ├── austin3_01452_334_335.png │ │ ├── austin4_00061_481_276.png │ │ ├── austin4_00133_264_308.png │ │ ├── austin4_00163_356_367.png │ │ ├── austin4_00205_468_478.png │ │ ├── austin7_00079_495_463.png │ │ ├── austin7_00100_300_263.png │ │ └── austin7_00166_430_319.png │ └── label │ │ ├── austin2_01179_258_289.png │ │ ├── austin3_00294_332_293.png │ │ ├── austin3_00497_432_361.png │ │ ├── austin3_01452_334_335.png │ │ ├── austin4_00061_481_276.png │ │ ├── austin4_00133_264_308.png │ │ ├── austin4_00163_356_367.png │ │ ├── austin4_00205_468_478.png │ │ ├── austin7_00079_495_463.png │ │ ├── austin7_00100_300_263.png │ │ └── austin7_00166_430_319.png └── val_samples │ ├── img │ ├── austin2_01179_258_289.png │ ├── austin3_00294_332_293.png │ ├── austin3_00497_432_361.png │ ├── austin3_01452_334_335.png │ ├── austin4_00061_481_276.png │ ├── austin4_00133_264_308.png │ ├── austin4_00163_356_367.png │ ├── austin4_00205_468_478.png │ ├── austin7_00079_495_463.png │ ├── austin7_00100_300_263.png │ └── austin7_00166_430_319.png │ └── label │ ├── austin2_01179_258_289.png │ ├── austin3_00294_332_293.png │ ├── austin3_00497_432_361.png │ ├── austin3_01452_334_335.png │ ├── austin4_00061_481_276.png │ ├── austin4_00133_264_308.png │ ├── austin4_00163_356_367.png │ ├── austin4_00205_468_478.png │ ├── austin7_00079_495_463.png │ ├── austin7_00100_300_263.png │ └── austin7_00166_430_319.png ├── LICENSE ├── Models └── BackBone │ ├── GetBackbone.py │ ├── ResNet.py │ ├── VGGNet.py │ └── __init__.py ├── README.md ├── README_zh-CN.md ├── STTNet.py ├── Test.py ├── Tools ├── CutImgSegWithLabel.py ├── GetImgMeanStd.py └── GetTrainValTestCSV.py ├── Train.py ├── Utils ├── Augmentations.py ├── Datasets.py └── Utils.py └── generate_dep_info ├── mean_std_info_test.json ├── test_data.csv ├── train_data.csv └── val_data.csv /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | __pycache__ 3 | .DS_Store 4 | 5 | *.pt 6 | Checkpoints 7 | Results 8 | Pretrain -------------------------------------------------------------------------------- /Data/test_samples/img/austin2_01179_258_289.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin2_01179_258_289.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin3_00294_332_293.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin3_00294_332_293.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin3_00497_432_361.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin3_00497_432_361.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin3_01452_334_335.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin3_01452_334_335.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin4_00061_481_276.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin4_00061_481_276.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin4_00133_264_308.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin4_00133_264_308.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin4_00163_356_367.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin4_00163_356_367.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin4_00205_468_478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin4_00205_468_478.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin7_00079_495_463.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin7_00079_495_463.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin7_00100_300_263.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin7_00100_300_263.png -------------------------------------------------------------------------------- /Data/test_samples/img/austin7_00166_430_319.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/img/austin7_00166_430_319.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin2_01179_258_289.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin2_01179_258_289.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin3_00294_332_293.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin3_00294_332_293.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin3_00497_432_361.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin3_00497_432_361.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin3_01452_334_335.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin3_01452_334_335.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin4_00061_481_276.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin4_00061_481_276.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin4_00133_264_308.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin4_00133_264_308.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin4_00163_356_367.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin4_00163_356_367.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin4_00205_468_478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin4_00205_468_478.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin7_00079_495_463.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin7_00079_495_463.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin7_00100_300_263.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin7_00100_300_263.png -------------------------------------------------------------------------------- /Data/test_samples/label/austin7_00166_430_319.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/test_samples/label/austin7_00166_430_319.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin2_01179_258_289.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin2_01179_258_289.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin3_00294_332_293.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin3_00294_332_293.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin3_00497_432_361.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin3_00497_432_361.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin3_01452_334_335.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin3_01452_334_335.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin4_00061_481_276.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin4_00061_481_276.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin4_00133_264_308.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin4_00133_264_308.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin4_00163_356_367.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin4_00163_356_367.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin4_00205_468_478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin4_00205_468_478.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin7_00079_495_463.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin7_00079_495_463.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin7_00100_300_263.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin7_00100_300_263.png -------------------------------------------------------------------------------- /Data/train_samples/img/austin7_00166_430_319.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/img/austin7_00166_430_319.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin2_01179_258_289.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin2_01179_258_289.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin3_00294_332_293.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin3_00294_332_293.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin3_00497_432_361.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin3_00497_432_361.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin3_01452_334_335.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin3_01452_334_335.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin4_00061_481_276.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin4_00061_481_276.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin4_00133_264_308.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin4_00133_264_308.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin4_00163_356_367.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin4_00163_356_367.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin4_00205_468_478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin4_00205_468_478.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin7_00079_495_463.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin7_00079_495_463.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin7_00100_300_263.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin7_00100_300_263.png -------------------------------------------------------------------------------- /Data/train_samples/label/austin7_00166_430_319.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/train_samples/label/austin7_00166_430_319.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin2_01179_258_289.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin2_01179_258_289.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin3_00294_332_293.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin3_00294_332_293.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin3_00497_432_361.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin3_00497_432_361.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin3_01452_334_335.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin3_01452_334_335.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin4_00061_481_276.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin4_00061_481_276.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin4_00133_264_308.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin4_00133_264_308.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin4_00163_356_367.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin4_00163_356_367.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin4_00205_468_478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin4_00205_468_478.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin7_00079_495_463.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin7_00079_495_463.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin7_00100_300_263.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin7_00100_300_263.png -------------------------------------------------------------------------------- /Data/val_samples/img/austin7_00166_430_319.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/img/austin7_00166_430_319.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin2_01179_258_289.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin2_01179_258_289.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin3_00294_332_293.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin3_00294_332_293.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin3_00497_432_361.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin3_00497_432_361.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin3_01452_334_335.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin3_01452_334_335.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin4_00061_481_276.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin4_00061_481_276.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin4_00133_264_308.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin4_00133_264_308.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin4_00163_356_367.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin4_00163_356_367.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin4_00205_468_478.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin4_00205_468_478.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin7_00079_495_463.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin7_00079_495_463.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin7_00100_300_263.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin7_00100_300_263.png -------------------------------------------------------------------------------- /Data/val_samples/label/austin7_00166_430_319.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Data/val_samples/label/austin7_00166_430_319.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/LICENSE -------------------------------------------------------------------------------- /Models/BackBone/GetBackbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Models/BackBone/GetBackbone.py -------------------------------------------------------------------------------- /Models/BackBone/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Models/BackBone/ResNet.py -------------------------------------------------------------------------------- /Models/BackBone/VGGNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Models/BackBone/VGGNet.py -------------------------------------------------------------------------------- /Models/BackBone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Models/BackBone/__init__.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/README.md -------------------------------------------------------------------------------- /README_zh-CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/README_zh-CN.md -------------------------------------------------------------------------------- /STTNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/STTNet.py -------------------------------------------------------------------------------- /Test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Test.py -------------------------------------------------------------------------------- /Tools/CutImgSegWithLabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Tools/CutImgSegWithLabel.py -------------------------------------------------------------------------------- /Tools/GetImgMeanStd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Tools/GetImgMeanStd.py -------------------------------------------------------------------------------- /Tools/GetTrainValTestCSV.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Tools/GetTrainValTestCSV.py -------------------------------------------------------------------------------- /Train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Train.py -------------------------------------------------------------------------------- /Utils/Augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Utils/Augmentations.py -------------------------------------------------------------------------------- /Utils/Datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Utils/Datasets.py -------------------------------------------------------------------------------- /Utils/Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/Utils/Utils.py -------------------------------------------------------------------------------- /generate_dep_info/mean_std_info_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/generate_dep_info/mean_std_info_test.json -------------------------------------------------------------------------------- /generate_dep_info/test_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/generate_dep_info/test_data.csv -------------------------------------------------------------------------------- /generate_dep_info/train_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/generate_dep_info/train_data.csv -------------------------------------------------------------------------------- /generate_dep_info/val_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KyanChen/STT/HEAD/generate_dep_info/val_data.csv --------------------------------------------------------------------------------