├── .idea ├── .gitignore ├── T-HOneCls.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── HOC ├── __init__.py ├── apis │ ├── __init__.py │ ├── base_trainer.py │ ├── build_trainer.py │ ├── self_calibration_trainer.py │ ├── single_model_trainer.py │ ├── toolkit_self_calibration │ │ ├── __init__.py │ │ ├── kl_loss.py │ │ └── update_ema_model.py │ └── validation.py ├── datasets │ ├── __init__.py │ ├── build_dataloader.py │ ├── data_base │ │ ├── __init__.py │ │ ├── base_dataset.py │ │ └── base_minibatchsampler.py │ ├── data_utils │ │ ├── __init__.py │ │ ├── data_preprocess.py │ │ ├── image_io.py │ │ ├── minibatch_sample.py │ │ └── train_test_separation.py │ ├── dataset_HanChuan.py │ ├── dataset_HongHu.py │ └── dataset_LongKou.py ├── loss_functions │ ├── __init__.py │ ├── build_loss_function.py │ ├── loss_functions.py │ ├── pf_oc_loss.py │ └── pf_vpu_loss.py ├── models │ ├── __init__.py │ ├── build_model.py │ ├── cbam.py │ └── freeocnet.py ├── optimization │ ├── __init__.py │ ├── build_lr_scheduler.py │ └── build_optimizer.py └── utils │ ├── __init__.py │ ├── build.py │ ├── classmap2rgbmap.py │ ├── logging_tool.py │ ├── logits_norm.py │ ├── metrics │ ├── __init__.py │ └── metrics.py │ ├── read_config.py │ ├── registry.py │ ├── scalar_recorder.py │ └── set_random_seed.py ├── README.md ├── configs ├── HanChuan │ ├── OCLoss │ │ ├── 1.py │ │ ├── 14.py │ │ ├── 16.py │ │ ├── 2.py │ │ ├── 3.py │ │ └── 6.py │ └── T-HOneCls │ │ ├── 1.py │ │ ├── 14.py │ │ ├── 16.py │ │ ├── 2.py │ │ ├── 3.py │ │ └── 6.py ├── HongHu │ ├── OCLoss │ │ ├── 10.py │ │ ├── 4.py │ │ ├── 6.py │ │ ├── 7.py │ │ └── 9.py │ └── T-HOneCls │ │ ├── 10.py │ │ ├── 4.py │ │ ├── 6.py │ │ ├── 7.py │ │ └── 9.py └── LongKou │ ├── OCLoss │ ├── 1.py │ ├── 3.py │ ├── 4.py │ └── 6.py │ └── T-HOneCls │ ├── 1.py │ ├── 3.py │ ├── 4.py │ └── 6.py ├── scripts ├── HanChuan.sh ├── HongHu.sh └── LongKou.sh └── tools └── train.py /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/T-HOneCls.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/.idea/T-HOneCls.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /HOC/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /HOC/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/__init__.py -------------------------------------------------------------------------------- /HOC/apis/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/base_trainer.py -------------------------------------------------------------------------------- /HOC/apis/build_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/build_trainer.py -------------------------------------------------------------------------------- /HOC/apis/self_calibration_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/self_calibration_trainer.py -------------------------------------------------------------------------------- /HOC/apis/single_model_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/single_model_trainer.py -------------------------------------------------------------------------------- /HOC/apis/toolkit_self_calibration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /HOC/apis/toolkit_self_calibration/kl_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/toolkit_self_calibration/kl_loss.py -------------------------------------------------------------------------------- /HOC/apis/toolkit_self_calibration/update_ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/toolkit_self_calibration/update_ema_model.py -------------------------------------------------------------------------------- /HOC/apis/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/apis/validation.py -------------------------------------------------------------------------------- /HOC/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/__init__.py -------------------------------------------------------------------------------- /HOC/datasets/build_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/build_dataloader.py -------------------------------------------------------------------------------- /HOC/datasets/data_base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_base/__init__.py -------------------------------------------------------------------------------- /HOC/datasets/data_base/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_base/base_dataset.py -------------------------------------------------------------------------------- /HOC/datasets/data_base/base_minibatchsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_base/base_minibatchsampler.py -------------------------------------------------------------------------------- /HOC/datasets/data_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_utils/__init__.py -------------------------------------------------------------------------------- /HOC/datasets/data_utils/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_utils/data_preprocess.py -------------------------------------------------------------------------------- /HOC/datasets/data_utils/image_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_utils/image_io.py -------------------------------------------------------------------------------- /HOC/datasets/data_utils/minibatch_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_utils/minibatch_sample.py -------------------------------------------------------------------------------- /HOC/datasets/data_utils/train_test_separation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/data_utils/train_test_separation.py -------------------------------------------------------------------------------- /HOC/datasets/dataset_HanChuan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/dataset_HanChuan.py -------------------------------------------------------------------------------- /HOC/datasets/dataset_HongHu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/dataset_HongHu.py -------------------------------------------------------------------------------- /HOC/datasets/dataset_LongKou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/datasets/dataset_LongKou.py -------------------------------------------------------------------------------- /HOC/loss_functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/loss_functions/__init__.py -------------------------------------------------------------------------------- /HOC/loss_functions/build_loss_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/loss_functions/build_loss_function.py -------------------------------------------------------------------------------- /HOC/loss_functions/loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/loss_functions/loss_functions.py -------------------------------------------------------------------------------- /HOC/loss_functions/pf_oc_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/loss_functions/pf_oc_loss.py -------------------------------------------------------------------------------- /HOC/loss_functions/pf_vpu_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/loss_functions/pf_vpu_loss.py -------------------------------------------------------------------------------- /HOC/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/models/__init__.py -------------------------------------------------------------------------------- /HOC/models/build_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/models/build_model.py -------------------------------------------------------------------------------- /HOC/models/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/models/cbam.py -------------------------------------------------------------------------------- /HOC/models/freeocnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/models/freeocnet.py -------------------------------------------------------------------------------- /HOC/optimization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/optimization/__init__.py -------------------------------------------------------------------------------- /HOC/optimization/build_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/optimization/build_lr_scheduler.py -------------------------------------------------------------------------------- /HOC/optimization/build_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/optimization/build_optimizer.py -------------------------------------------------------------------------------- /HOC/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/__init__.py -------------------------------------------------------------------------------- /HOC/utils/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/build.py -------------------------------------------------------------------------------- /HOC/utils/classmap2rgbmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/classmap2rgbmap.py -------------------------------------------------------------------------------- /HOC/utils/logging_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/logging_tool.py -------------------------------------------------------------------------------- /HOC/utils/logits_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/logits_norm.py -------------------------------------------------------------------------------- /HOC/utils/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /HOC/utils/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/metrics/metrics.py -------------------------------------------------------------------------------- /HOC/utils/read_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/read_config.py -------------------------------------------------------------------------------- /HOC/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/registry.py -------------------------------------------------------------------------------- /HOC/utils/scalar_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/scalar_recorder.py -------------------------------------------------------------------------------- /HOC/utils/set_random_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/HOC/utils/set_random_seed.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/README.md -------------------------------------------------------------------------------- /configs/HanChuan/OCLoss/1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/OCLoss/1.py -------------------------------------------------------------------------------- /configs/HanChuan/OCLoss/14.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/OCLoss/14.py -------------------------------------------------------------------------------- /configs/HanChuan/OCLoss/16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/OCLoss/16.py -------------------------------------------------------------------------------- /configs/HanChuan/OCLoss/2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/OCLoss/2.py -------------------------------------------------------------------------------- /configs/HanChuan/OCLoss/3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/OCLoss/3.py -------------------------------------------------------------------------------- /configs/HanChuan/OCLoss/6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/OCLoss/6.py -------------------------------------------------------------------------------- /configs/HanChuan/T-HOneCls/1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/T-HOneCls/1.py -------------------------------------------------------------------------------- /configs/HanChuan/T-HOneCls/14.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/T-HOneCls/14.py -------------------------------------------------------------------------------- /configs/HanChuan/T-HOneCls/16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/T-HOneCls/16.py -------------------------------------------------------------------------------- /configs/HanChuan/T-HOneCls/2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/T-HOneCls/2.py -------------------------------------------------------------------------------- /configs/HanChuan/T-HOneCls/3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/T-HOneCls/3.py -------------------------------------------------------------------------------- /configs/HanChuan/T-HOneCls/6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HanChuan/T-HOneCls/6.py -------------------------------------------------------------------------------- /configs/HongHu/OCLoss/10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/OCLoss/10.py -------------------------------------------------------------------------------- /configs/HongHu/OCLoss/4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/OCLoss/4.py -------------------------------------------------------------------------------- /configs/HongHu/OCLoss/6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/OCLoss/6.py -------------------------------------------------------------------------------- /configs/HongHu/OCLoss/7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/OCLoss/7.py -------------------------------------------------------------------------------- /configs/HongHu/OCLoss/9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/OCLoss/9.py -------------------------------------------------------------------------------- /configs/HongHu/T-HOneCls/10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/T-HOneCls/10.py -------------------------------------------------------------------------------- /configs/HongHu/T-HOneCls/4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/T-HOneCls/4.py -------------------------------------------------------------------------------- /configs/HongHu/T-HOneCls/6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/T-HOneCls/6.py -------------------------------------------------------------------------------- /configs/HongHu/T-HOneCls/7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/T-HOneCls/7.py -------------------------------------------------------------------------------- /configs/HongHu/T-HOneCls/9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/HongHu/T-HOneCls/9.py -------------------------------------------------------------------------------- /configs/LongKou/OCLoss/1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/OCLoss/1.py -------------------------------------------------------------------------------- /configs/LongKou/OCLoss/3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/OCLoss/3.py -------------------------------------------------------------------------------- /configs/LongKou/OCLoss/4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/OCLoss/4.py -------------------------------------------------------------------------------- /configs/LongKou/OCLoss/6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/OCLoss/6.py -------------------------------------------------------------------------------- /configs/LongKou/T-HOneCls/1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/T-HOneCls/1.py -------------------------------------------------------------------------------- /configs/LongKou/T-HOneCls/3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/T-HOneCls/3.py -------------------------------------------------------------------------------- /configs/LongKou/T-HOneCls/4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/T-HOneCls/4.py -------------------------------------------------------------------------------- /configs/LongKou/T-HOneCls/6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/configs/LongKou/T-HOneCls/6.py -------------------------------------------------------------------------------- /scripts/HanChuan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/scripts/HanChuan.sh -------------------------------------------------------------------------------- /scripts/HongHu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/scripts/HongHu.sh -------------------------------------------------------------------------------- /scripts/LongKou.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/scripts/LongKou.sh -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/T-HOneCls/HEAD/tools/train.py --------------------------------------------------------------------------------