├── .gitignore ├── README.md ├── configs ├── __base__ │ ├── accelerate.py │ └── newbase.py ├── dataset │ └── cod4040.py └── uscod │ ├── CORAL_dinov1.py │ ├── CORAL_dinov2.py │ ├── UCOD-DPL_dinov1.py │ └── UCOD-DPL_dinov2.py ├── data ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── cache_manager.py │ ├── dataloader_utils.py │ ├── lr_dataset.py │ ├── transforms.py │ └── uscod_dataset.py └── utils │ ├── __init__.py │ ├── feature_extractor.py │ └── found_bkg_mask.py ├── engine ├── __init__.py ├── config │ ├── __init__.py │ ├── base_cfg.py │ └── config.py ├── registry │ ├── __init__.py │ ├── build_function.py │ ├── registry.py │ └── root.py ├── runner │ ├── loop_CORAL.py │ ├── loop_UCOD_DPL.py │ ├── runner.py │ └── utils.py └── utils │ ├── bilateral_solver.py │ ├── fileio │ ├── backend │ │ ├── __init__.py │ │ ├── filetype │ │ │ ├── __init__.py │ │ │ ├── base_type.py │ │ │ ├── image_type.py │ │ │ ├── json_type.py │ │ │ └── pickle_type.py │ │ └── ioctl │ │ │ ├── __init__.py │ │ │ ├── baseio.py │ │ │ ├── imageio.py │ │ │ ├── jsonio.py │ │ │ ├── pickleio.py │ │ │ └── utils.py │ └── cache.py │ ├── logger.py │ ├── metrics │ └── metric.py │ ├── misc.py │ ├── save_image.py │ ├── seed.py │ └── show_imgs.py ├── generate_pseudo_label.py ├── images ├── performance_CORAL.png └── performance_UCOD-DPL.png ├── models ├── UDLR.py ├── __init__.py ├── backbones │ ├── build_backbone.py │ ├── dino.py │ ├── pvt_v2.py │ ├── res2net_v1b.py │ ├── resnet.py │ ├── swin_v1.py │ └── vit.py ├── discriminator.py ├── modules │ ├── ASR.py │ ├── CSF.py │ ├── DBA.py │ ├── GE_pix_level.py │ ├── HRE.py │ ├── attention.py │ ├── full_model.py │ ├── misc.py │ └── mlp.py └── uscod.py ├── requirement.txt ├── scripts ├── LTeval.py ├── LTtrain.py ├── args.py ├── eval.py ├── launch_train_first_stage.sh ├── launch_train_second_stage.sh ├── launch_val_first_stage.sh ├── launch_val_second_stage.sh └── train.py └── weights ├── CORAL_dinov1.safetensors ├── CORAL_dinov2.safetensors ├── UCOD_DPL_dinov1.safetensors └── UCOD_DPL_dinov2.safetensors /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/README.md -------------------------------------------------------------------------------- /configs/__base__/accelerate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/__base__/accelerate.py -------------------------------------------------------------------------------- /configs/__base__/newbase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/__base__/newbase.py -------------------------------------------------------------------------------- /configs/dataset/cod4040.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/dataset/cod4040.py -------------------------------------------------------------------------------- /configs/uscod/CORAL_dinov1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/uscod/CORAL_dinov1.py -------------------------------------------------------------------------------- /configs/uscod/CORAL_dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/uscod/CORAL_dinov2.py -------------------------------------------------------------------------------- /configs/uscod/UCOD-DPL_dinov1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/uscod/UCOD-DPL_dinov1.py -------------------------------------------------------------------------------- /configs/uscod/UCOD-DPL_dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/configs/uscod/UCOD-DPL_dinov2.py -------------------------------------------------------------------------------- /data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/__init__.py -------------------------------------------------------------------------------- /data/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/base_dataset.py -------------------------------------------------------------------------------- /data/datasets/cache_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/cache_manager.py -------------------------------------------------------------------------------- /data/datasets/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/dataloader_utils.py -------------------------------------------------------------------------------- /data/datasets/lr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/lr_dataset.py -------------------------------------------------------------------------------- /data/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/transforms.py -------------------------------------------------------------------------------- /data/datasets/uscod_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/datasets/uscod_dataset.py -------------------------------------------------------------------------------- /data/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/utils/feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/utils/feature_extractor.py -------------------------------------------------------------------------------- /data/utils/found_bkg_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/data/utils/found_bkg_mask.py -------------------------------------------------------------------------------- /engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engine/config/__init__.py: -------------------------------------------------------------------------------- 1 | from .config import CfgNode 2 | 3 | -------------------------------------------------------------------------------- /engine/config/base_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/config/base_cfg.py -------------------------------------------------------------------------------- /engine/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/config/config.py -------------------------------------------------------------------------------- /engine/registry/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | from .root import * 3 | 4 | -------------------------------------------------------------------------------- /engine/registry/build_function.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engine/registry/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/registry/registry.py -------------------------------------------------------------------------------- /engine/registry/root.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/registry/root.py -------------------------------------------------------------------------------- /engine/runner/loop_CORAL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/runner/loop_CORAL.py -------------------------------------------------------------------------------- /engine/runner/loop_UCOD_DPL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/runner/loop_UCOD_DPL.py -------------------------------------------------------------------------------- /engine/runner/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/runner/runner.py -------------------------------------------------------------------------------- /engine/runner/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/runner/utils.py -------------------------------------------------------------------------------- /engine/utils/bilateral_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/bilateral_solver.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/__init__.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/filetype/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/filetype/__init__.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/filetype/base_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/filetype/base_type.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/filetype/image_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/filetype/image_type.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/filetype/json_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/filetype/json_type.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/filetype/pickle_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/filetype/pickle_type.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/ioctl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/ioctl/__init__.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/ioctl/baseio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/ioctl/baseio.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/ioctl/imageio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/ioctl/imageio.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/ioctl/jsonio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/ioctl/jsonio.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/ioctl/pickleio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/ioctl/pickleio.py -------------------------------------------------------------------------------- /engine/utils/fileio/backend/ioctl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/fileio/backend/ioctl/utils.py -------------------------------------------------------------------------------- /engine/utils/fileio/cache.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engine/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/logger.py -------------------------------------------------------------------------------- /engine/utils/metrics/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/metrics/metric.py -------------------------------------------------------------------------------- /engine/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/misc.py -------------------------------------------------------------------------------- /engine/utils/save_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/save_image.py -------------------------------------------------------------------------------- /engine/utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/seed.py -------------------------------------------------------------------------------- /engine/utils/show_imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/engine/utils/show_imgs.py -------------------------------------------------------------------------------- /generate_pseudo_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/generate_pseudo_label.py -------------------------------------------------------------------------------- /images/performance_CORAL.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/images/performance_CORAL.png -------------------------------------------------------------------------------- /images/performance_UCOD-DPL.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/images/performance_UCOD-DPL.png -------------------------------------------------------------------------------- /models/UDLR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/UDLR.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbones/build_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/build_backbone.py -------------------------------------------------------------------------------- /models/backbones/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/dino.py -------------------------------------------------------------------------------- /models/backbones/pvt_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/pvt_v2.py -------------------------------------------------------------------------------- /models/backbones/res2net_v1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/res2net_v1b.py -------------------------------------------------------------------------------- /models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/resnet.py -------------------------------------------------------------------------------- /models/backbones/swin_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/swin_v1.py -------------------------------------------------------------------------------- /models/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/backbones/vit.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/modules/ASR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/ASR.py -------------------------------------------------------------------------------- /models/modules/CSF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/CSF.py -------------------------------------------------------------------------------- /models/modules/DBA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/DBA.py -------------------------------------------------------------------------------- /models/modules/GE_pix_level.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/GE_pix_level.py -------------------------------------------------------------------------------- /models/modules/HRE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/HRE.py -------------------------------------------------------------------------------- /models/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/attention.py -------------------------------------------------------------------------------- /models/modules/full_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/full_model.py -------------------------------------------------------------------------------- /models/modules/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/misc.py -------------------------------------------------------------------------------- /models/modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/modules/mlp.py -------------------------------------------------------------------------------- /models/uscod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/models/uscod.py -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/requirement.txt -------------------------------------------------------------------------------- /scripts/LTeval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/LTeval.py -------------------------------------------------------------------------------- /scripts/LTtrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/LTtrain.py -------------------------------------------------------------------------------- /scripts/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/args.py -------------------------------------------------------------------------------- /scripts/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/eval.py -------------------------------------------------------------------------------- /scripts/launch_train_first_stage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/launch_train_first_stage.sh -------------------------------------------------------------------------------- /scripts/launch_train_second_stage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/launch_train_second_stage.sh -------------------------------------------------------------------------------- /scripts/launch_val_first_stage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/launch_val_first_stage.sh -------------------------------------------------------------------------------- /scripts/launch_val_second_stage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/launch_val_second_stage.sh -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/scripts/train.py -------------------------------------------------------------------------------- /weights/CORAL_dinov1.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/weights/CORAL_dinov1.safetensors -------------------------------------------------------------------------------- /weights/CORAL_dinov2.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/weights/CORAL_dinov2.safetensors -------------------------------------------------------------------------------- /weights/UCOD_DPL_dinov1.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/weights/UCOD_DPL_dinov1.safetensors -------------------------------------------------------------------------------- /weights/UCOD_DPL_dinov2.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Heartfirey/UCOD-DPL/HEAD/weights/UCOD_DPL_dinov2.safetensors --------------------------------------------------------------------------------