├── README.md ├── core ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-35.pyc │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-38.pyc │ ├── base.cpython-35.pyc │ ├── base.cpython-36.pyc │ ├── base.cpython-37.pyc │ ├── base.cpython-38.pyc │ ├── test.cpython-38.pyc │ ├── train.cpython-36.pyc │ ├── train.cpython-37.pyc │ ├── train.cpython-38.pyc │ ├── train_heatmap_no_grad_reasoning.cpython-35.pyc │ ├── train_heatmapmask_id_reasoning.cpython-35.pyc │ ├── train_heatmapmask_no_grad_reasoning.cpython-35.pyc │ ├── train_heatmapmask_reasoning.cpython-35.pyc │ └── train_with_cam.cpython-35.pyc ├── base.py ├── test.py └── train.py ├── data_loader ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── dataset.cpython-38.pyc │ ├── loader.cpython-38.pyc │ ├── processing.cpython-38.pyc │ └── sampler.cpython-38.pyc ├── dataset.py ├── loader.py ├── processing.py └── sampler.py ├── main.py ├── network ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-35.pyc │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-38.pyc │ ├── gem_pool.cpython-38.pyc │ ├── lr.cpython-38.pyc │ ├── model.cpython-36.pyc │ ├── model.cpython-37.pyc │ └── model.cpython-38.pyc ├── clip │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── clip.cpython-38.pyc │ │ ├── model.cpython-38.pyc │ │ └── simple_tokenizer.cpython-38.pyc │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py ├── gem_pool.py ├── lr.py ├── model.py └── processing.py ├── run.sh └── tools ├── MSEL.py ├── __init__.py ├── __pycache__ ├── __init__.cpython-35.pyc ├── __init__.cpython-36.pyc ├── __init__.cpython-37.pyc ├── __init__.cpython-38.pyc ├── eval_metrics.cpython-38.pyc ├── logger.cpython-35.pyc ├── logger.cpython-36.pyc ├── logger.cpython-37.pyc ├── logger.cpython-38.pyc ├── loss.cpython-35.pyc ├── loss.cpython-36.pyc ├── loss.cpython-37.pyc ├── loss.cpython-38.pyc ├── meter.cpython-35.pyc ├── meter.cpython-36.pyc ├── meter.cpython-37.pyc ├── meter.cpython-38.pyc ├── utils.cpython-35.pyc ├── utils.cpython-36.pyc ├── utils.cpython-37.pyc └── utils.cpython-38.pyc ├── eval_metrics.py ├── logger.py ├── loss.py ├── meter.py └── utils.py /README.md: -------------------------------------------------------------------------------- 1 | 权重已放出。链接：https://pan.baidu.com/s/1kiMj34iGZMvxFrXmZ4d9EA 2 | 提取码：b7l1 3 | -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__init__.py -------------------------------------------------------------------------------- /core/__pycache__/__init__.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/__init__.cpython-35.pyc -------------------------------------------------------------------------------- /core/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /core/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /core/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /core/__pycache__/base.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/base.cpython-35.pyc -------------------------------------------------------------------------------- /core/__pycache__/base.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/base.cpython-36.pyc -------------------------------------------------------------------------------- /core/__pycache__/base.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/base.cpython-37.pyc -------------------------------------------------------------------------------- /core/__pycache__/base.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/base.cpython-38.pyc -------------------------------------------------------------------------------- /core/__pycache__/test.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/test.cpython-38.pyc -------------------------------------------------------------------------------- /core/__pycache__/train.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train.cpython-36.pyc -------------------------------------------------------------------------------- /core/__pycache__/train.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train.cpython-37.pyc -------------------------------------------------------------------------------- /core/__pycache__/train.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train.cpython-38.pyc -------------------------------------------------------------------------------- /core/__pycache__/train_heatmap_no_grad_reasoning.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train_heatmap_no_grad_reasoning.cpython-35.pyc -------------------------------------------------------------------------------- /core/__pycache__/train_heatmapmask_id_reasoning.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train_heatmapmask_id_reasoning.cpython-35.pyc -------------------------------------------------------------------------------- /core/__pycache__/train_heatmapmask_no_grad_reasoning.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train_heatmapmask_no_grad_reasoning.cpython-35.pyc -------------------------------------------------------------------------------- /core/__pycache__/train_heatmapmask_reasoning.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train_heatmapmask_reasoning.cpython-35.pyc -------------------------------------------------------------------------------- /core/__pycache__/train_with_cam.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/__pycache__/train_with_cam.cpython-35.pyc -------------------------------------------------------------------------------- /core/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/base.py -------------------------------------------------------------------------------- /core/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/test.py -------------------------------------------------------------------------------- /core/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/core/train.py -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_loader/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /data_loader/__pycache__/dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/__pycache__/dataset.cpython-38.pyc -------------------------------------------------------------------------------- /data_loader/__pycache__/loader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/__pycache__/loader.cpython-38.pyc -------------------------------------------------------------------------------- /data_loader/__pycache__/processing.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/__pycache__/processing.cpython-38.pyc -------------------------------------------------------------------------------- /data_loader/__pycache__/sampler.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/__pycache__/sampler.cpython-38.pyc -------------------------------------------------------------------------------- /data_loader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/dataset.py -------------------------------------------------------------------------------- /data_loader/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/loader.py -------------------------------------------------------------------------------- /data_loader/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/processing.py -------------------------------------------------------------------------------- /data_loader/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/data_loader/sampler.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/main.py -------------------------------------------------------------------------------- /network/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import * -------------------------------------------------------------------------------- /network/__pycache__/__init__.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/__init__.cpython-35.pyc -------------------------------------------------------------------------------- /network/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /network/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /network/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /network/__pycache__/gem_pool.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/gem_pool.cpython-38.pyc -------------------------------------------------------------------------------- /network/__pycache__/lr.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/lr.cpython-38.pyc -------------------------------------------------------------------------------- /network/__pycache__/model.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/model.cpython-36.pyc -------------------------------------------------------------------------------- /network/__pycache__/model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/model.cpython-37.pyc -------------------------------------------------------------------------------- /network/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /network/clip/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | from .clip import * -------------------------------------------------------------------------------- /network/clip/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /network/clip/__pycache__/clip.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/__pycache__/clip.cpython-38.pyc -------------------------------------------------------------------------------- /network/clip/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /network/clip/__pycache__/simple_tokenizer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/__pycache__/simple_tokenizer.cpython-38.pyc -------------------------------------------------------------------------------- /network/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /network/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/clip.py -------------------------------------------------------------------------------- /network/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/model.py -------------------------------------------------------------------------------- /network/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /network/gem_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/gem_pool.py -------------------------------------------------------------------------------- /network/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/lr.py -------------------------------------------------------------------------------- /network/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/model.py -------------------------------------------------------------------------------- /network/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/network/processing.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 python3 main.py --output_path sysu/base -------------------------------------------------------------------------------- /tools/MSEL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/MSEL.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/__pycache__/__init__.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/__init__.cpython-35.pyc -------------------------------------------------------------------------------- /tools/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /tools/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /tools/__pycache__/eval_metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/eval_metrics.cpython-38.pyc -------------------------------------------------------------------------------- /tools/__pycache__/logger.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/logger.cpython-35.pyc -------------------------------------------------------------------------------- /tools/__pycache__/logger.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/logger.cpython-36.pyc -------------------------------------------------------------------------------- /tools/__pycache__/logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/logger.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /tools/__pycache__/loss.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/loss.cpython-35.pyc -------------------------------------------------------------------------------- /tools/__pycache__/loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/loss.cpython-36.pyc -------------------------------------------------------------------------------- /tools/__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/loss.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/loss.cpython-38.pyc -------------------------------------------------------------------------------- /tools/__pycache__/meter.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/meter.cpython-35.pyc -------------------------------------------------------------------------------- /tools/__pycache__/meter.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/meter.cpython-36.pyc -------------------------------------------------------------------------------- /tools/__pycache__/meter.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/meter.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/meter.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/meter.cpython-38.pyc -------------------------------------------------------------------------------- /tools/__pycache__/utils.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/utils.cpython-35.pyc -------------------------------------------------------------------------------- /tools/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /tools/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /tools/eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/eval_metrics.py -------------------------------------------------------------------------------- /tools/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/logger.py -------------------------------------------------------------------------------- /tools/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/loss.py -------------------------------------------------------------------------------- /tools/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/meter.py -------------------------------------------------------------------------------- /tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nengdong96/CSDN/HEAD/tools/utils.py --------------------------------------------------------------------------------