├── .idea ├── .gitignore ├── HOneCls.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── README.md ├── configs ├── config_HanChuan.py ├── config_HongHu.py └── config_LongKou.py ├── data ├── __init__.py ├── data_base │ ├── __init__.py │ ├── base_dataset.py │ ├── base_minibatchsampler.py │ └── patch_based_dataset.py ├── data_utils │ ├── __init__.py │ ├── data_preprocess.py │ ├── minibatch_sample.py │ └── train_test_separation.py ├── dataloader.py ├── dataset_HanChuan.py ├── dataset_HongHu.py └── dataset_LongKou.py ├── main.py ├── model ├── __init__.py ├── cbam.py └── freeocnet.py ├── risk_estimation ├── __init__.py ├── absPU.py ├── loss_function.py └── one_class_risk_estimator.py ├── sh ├── HanChuan.sh ├── HongHu.sh └── LongKou.sh └── utils ├── __init__.py ├── classmap_2_RGBmap.py ├── get_cfg_data.py ├── logging_tool.py ├── metric.py ├── scalar_recorder.py └── set_random_seed.py /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/HOneCls.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/.idea/HOneCls.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/README.md -------------------------------------------------------------------------------- /configs/config_HanChuan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/configs/config_HanChuan.py -------------------------------------------------------------------------------- /configs/config_HongHu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/configs/config_HongHu.py -------------------------------------------------------------------------------- /configs/config_LongKou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/configs/config_LongKou.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/data_base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/data_base/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/data_base/base_dataset.py -------------------------------------------------------------------------------- /data/data_base/base_minibatchsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/data_base/base_minibatchsampler.py -------------------------------------------------------------------------------- /data/data_base/patch_based_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/data_base/patch_based_dataset.py -------------------------------------------------------------------------------- /data/data_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/data_utils/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/data_utils/data_preprocess.py -------------------------------------------------------------------------------- /data/data_utils/minibatch_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/data_utils/minibatch_sample.py -------------------------------------------------------------------------------- /data/data_utils/train_test_separation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/data_utils/train_test_separation.py -------------------------------------------------------------------------------- /data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/dataloader.py -------------------------------------------------------------------------------- /data/dataset_HanChuan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/dataset_HanChuan.py -------------------------------------------------------------------------------- /data/dataset_HongHu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/dataset_HongHu.py -------------------------------------------------------------------------------- /data/dataset_LongKou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/data/dataset_LongKou.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/model/cbam.py -------------------------------------------------------------------------------- /model/freeocnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/model/freeocnet.py -------------------------------------------------------------------------------- /risk_estimation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/risk_estimation/__init__.py -------------------------------------------------------------------------------- /risk_estimation/absPU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/risk_estimation/absPU.py -------------------------------------------------------------------------------- /risk_estimation/loss_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/risk_estimation/loss_function.py -------------------------------------------------------------------------------- /risk_estimation/one_class_risk_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/risk_estimation/one_class_risk_estimator.py -------------------------------------------------------------------------------- /sh/HanChuan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/sh/HanChuan.sh -------------------------------------------------------------------------------- /sh/HongHu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/sh/HongHu.sh -------------------------------------------------------------------------------- /sh/LongKou.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/sh/LongKou.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/classmap_2_RGBmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/classmap_2_RGBmap.py -------------------------------------------------------------------------------- /utils/get_cfg_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/get_cfg_data.py -------------------------------------------------------------------------------- /utils/logging_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/logging_tool.py -------------------------------------------------------------------------------- /utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/metric.py -------------------------------------------------------------------------------- /utils/scalar_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/scalar_recorder.py -------------------------------------------------------------------------------- /utils/set_random_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Hengwei-Zhao96/HOneCls/HEAD/utils/set_random_seed.py --------------------------------------------------------------------------------