├── DepthBranchDecoder ├── DepthBranchDecoder.py ├── __init__.py └── __pycache__ │ ├── DepthBranchDecoder.cpython-36.pyc │ ├── FlowBranchDecoder.cpython-36.pyc │ └── __init__.cpython-36.pyc ├── DepthBranchEncoder ├── DepthBranchEncoder.py ├── __init__.py └── __pycache__ │ ├── DepthBranchEncoder.cpython-36.pyc │ ├── FlowBranchEncoder.cpython-36.pyc │ └── __init__.cpython-36.pyc ├── ImageBranchDecoder ├── ImageBranchDecoder.py ├── __init__.py └── __pycache__ │ ├── ImageBranchDecoder.cpython-36.pyc │ ├── ImageBranchDecoder_parts.cpython-36.pyc │ └── __init__.cpython-36.pyc ├── ImageBranchEncoder ├── ImageBranchEncoder.py ├── __init__.py └── __pycache__ │ ├── ImageBranchEncoder.cpython-36.pyc │ └── __init__.cpython-36.pyc ├── ImageDepthNet ├── ImageDepthNet.py ├── __init__.py └── __pycache__ │ ├── ImageDepthNet.cpython-36.pyc │ ├── ImageFlowNet.cpython-36.pyc │ ├── ImageFlowNet_parts.cpython-36.pyc │ └── __init__.cpython-36.pyc ├── README.md ├── RGBdDataset_processed └── NLPR │ └── testset │ └── depth │ ├── 10_01-00-59.bmp │ ├── 10_01-02-01.bmp │ ├── 10_01-05-11.bmp │ ├── 10_01-06-39.bmp │ ├── 10_01-10-53.bmp │ ├── 10_01-11-07.bmp │ ├── 10_01-13-37.bmp │ ├── 10_01-16-00.bmp │ ├── 10_01-16-04.bmp │ ├── 10_01-16-21.bmp │ ├── 10_01-17-25.bmp │ ├── 10_01-17-42.bmp │ ├── 10_01-18-10.bmp │ ├── 10_01-18-15.bmp │ ├── 10_01-20-50.bmp │ ├── 10_01-21-12.bmp │ ├── 10_01-21-21.bmp │ ├── 10_01-21-41.bmp │ ├── 10_01-22-41.bmp │ ├── 10_01-30-24.bmp │ ├── 10_02-58-26.bmp │ └── 10_02-58-46.bmp ├── dataset.py ├── finetune_DUT_RGBD.py ├── generate_list.py ├── list ├── test │ └── test_list.txt └── train │ ├── DUT_train_list.txt │ └── train_list.txt ├── model_epoch_loss └── loss.txt ├── output └── NLPR │ └── S2MA.pth │ ├── 1_02-03-44.png │ ├── 1_02-08-35.png │ ├── 1_02-54-18.png │ └── 1_02-59-20.png ├── parameter.py ├── parameter_finetune_DUT_RGBD.py ├── test.py ├── train.py └── transforms.py /DepthBranchDecoder/DepthBranchDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchDecoder/DepthBranchDecoder.py -------------------------------------------------------------------------------- /DepthBranchDecoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchDecoder/__init__.py -------------------------------------------------------------------------------- /DepthBranchDecoder/__pycache__/DepthBranchDecoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchDecoder/__pycache__/DepthBranchDecoder.cpython-36.pyc -------------------------------------------------------------------------------- /DepthBranchDecoder/__pycache__/FlowBranchDecoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchDecoder/__pycache__/FlowBranchDecoder.cpython-36.pyc -------------------------------------------------------------------------------- /DepthBranchDecoder/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchDecoder/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /DepthBranchEncoder/DepthBranchEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchEncoder/DepthBranchEncoder.py -------------------------------------------------------------------------------- /DepthBranchEncoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchEncoder/__init__.py -------------------------------------------------------------------------------- /DepthBranchEncoder/__pycache__/DepthBranchEncoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchEncoder/__pycache__/DepthBranchEncoder.cpython-36.pyc -------------------------------------------------------------------------------- /DepthBranchEncoder/__pycache__/FlowBranchEncoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchEncoder/__pycache__/FlowBranchEncoder.cpython-36.pyc -------------------------------------------------------------------------------- /DepthBranchEncoder/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/DepthBranchEncoder/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /ImageBranchDecoder/ImageBranchDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchDecoder/ImageBranchDecoder.py -------------------------------------------------------------------------------- /ImageBranchDecoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchDecoder/__init__.py -------------------------------------------------------------------------------- /ImageBranchDecoder/__pycache__/ImageBranchDecoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchDecoder/__pycache__/ImageBranchDecoder.cpython-36.pyc -------------------------------------------------------------------------------- /ImageBranchDecoder/__pycache__/ImageBranchDecoder_parts.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchDecoder/__pycache__/ImageBranchDecoder_parts.cpython-36.pyc -------------------------------------------------------------------------------- /ImageBranchDecoder/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchDecoder/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /ImageBranchEncoder/ImageBranchEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchEncoder/ImageBranchEncoder.py -------------------------------------------------------------------------------- /ImageBranchEncoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchEncoder/__init__.py -------------------------------------------------------------------------------- /ImageBranchEncoder/__pycache__/ImageBranchEncoder.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchEncoder/__pycache__/ImageBranchEncoder.cpython-36.pyc -------------------------------------------------------------------------------- /ImageBranchEncoder/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageBranchEncoder/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /ImageDepthNet/ImageDepthNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageDepthNet/ImageDepthNet.py -------------------------------------------------------------------------------- /ImageDepthNet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageDepthNet/__init__.py -------------------------------------------------------------------------------- /ImageDepthNet/__pycache__/ImageDepthNet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageDepthNet/__pycache__/ImageDepthNet.cpython-36.pyc -------------------------------------------------------------------------------- /ImageDepthNet/__pycache__/ImageFlowNet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageDepthNet/__pycache__/ImageFlowNet.cpython-36.pyc -------------------------------------------------------------------------------- /ImageDepthNet/__pycache__/ImageFlowNet_parts.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageDepthNet/__pycache__/ImageFlowNet_parts.cpython-36.pyc -------------------------------------------------------------------------------- /ImageDepthNet/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/ImageDepthNet/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/README.md -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-00-59.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-00-59.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-02-01.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-02-01.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-05-11.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-05-11.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-06-39.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-06-39.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-10-53.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-10-53.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-11-07.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-11-07.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-13-37.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-13-37.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-16-00.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-16-00.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-16-04.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-16-04.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-16-21.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-16-21.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-17-25.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-17-25.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-17-42.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-17-42.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-18-10.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-18-10.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-18-15.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-18-15.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-20-50.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-20-50.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-21-12.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-21-12.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-21-21.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-21-21.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-21-41.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-21-41.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-22-41.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-22-41.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_01-30-24.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_01-30-24.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_02-58-26.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_02-58-26.bmp -------------------------------------------------------------------------------- /RGBdDataset_processed/NLPR/testset/depth/10_02-58-46.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/RGBdDataset_processed/NLPR/testset/depth/10_02-58-46.bmp -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/dataset.py -------------------------------------------------------------------------------- /finetune_DUT_RGBD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/finetune_DUT_RGBD.py -------------------------------------------------------------------------------- /generate_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/generate_list.py -------------------------------------------------------------------------------- /list/test/test_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/list/test/test_list.txt -------------------------------------------------------------------------------- /list/train/DUT_train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/list/train/DUT_train_list.txt -------------------------------------------------------------------------------- /list/train/train_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/list/train/train_list.txt -------------------------------------------------------------------------------- /model_epoch_loss/loss.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/model_epoch_loss/loss.txt -------------------------------------------------------------------------------- /output/NLPR/S2MA.pth/1_02-03-44.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/output/NLPR/S2MA.pth/1_02-03-44.png -------------------------------------------------------------------------------- /output/NLPR/S2MA.pth/1_02-08-35.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/output/NLPR/S2MA.pth/1_02-08-35.png -------------------------------------------------------------------------------- /output/NLPR/S2MA.pth/1_02-54-18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/output/NLPR/S2MA.pth/1_02-54-18.png -------------------------------------------------------------------------------- /output/NLPR/S2MA.pth/1_02-59-20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/output/NLPR/S2MA.pth/1_02-59-20.png -------------------------------------------------------------------------------- /parameter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/parameter.py -------------------------------------------------------------------------------- /parameter_finetune_DUT_RGBD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/parameter_finetune_DUT_RGBD.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/train.py -------------------------------------------------------------------------------- /transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nnizhang/S2MA/HEAD/transforms.py --------------------------------------------------------------------------------