├── .gitignore ├── README.md ├── codes ├── configs │ ├── init_neurop.yaml │ ├── test │ │ ├── test_neurop_mit5k_dark.yaml │ │ ├── test_neurop_mit5k_lite.yaml │ │ ├── test_neurop_ppr10k_a.yaml │ │ ├── test_neurop_ppr10k_b.yaml │ │ └── test_neurop_ppr10k_c.yaml │ └── train │ │ ├── train_neurop_mit5k_dark.yaml │ │ ├── train_neurop_mit5k_lite.yaml │ │ ├── train_neurop_ppr10k_a.yaml │ │ ├── train_neurop_ppr10k_b.yaml │ │ └── train_neurop_ppr10k_c.yaml ├── data │ ├── __init__.py │ └── dataset.py ├── models │ ├── __init__.py │ ├── loss.py │ ├── model.py │ └── networks.py ├── test.py ├── train.py └── utils.py ├── pretrained_models ├── neurop_fivek_dark.pth ├── neurop_fivek_lite.pth ├── neurop_ppr_a.pth ├── neurop_ppr_b.pth └── neurop_ppr_c.pth └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /codes/configs/init_neurop.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/init_neurop.yaml -------------------------------------------------------------------------------- /codes/configs/test/test_neurop_mit5k_dark.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/test/test_neurop_mit5k_dark.yaml -------------------------------------------------------------------------------- /codes/configs/test/test_neurop_mit5k_lite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/test/test_neurop_mit5k_lite.yaml -------------------------------------------------------------------------------- /codes/configs/test/test_neurop_ppr10k_a.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/test/test_neurop_ppr10k_a.yaml -------------------------------------------------------------------------------- /codes/configs/test/test_neurop_ppr10k_b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/test/test_neurop_ppr10k_b.yaml -------------------------------------------------------------------------------- /codes/configs/test/test_neurop_ppr10k_c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/test/test_neurop_ppr10k_c.yaml -------------------------------------------------------------------------------- /codes/configs/train/train_neurop_mit5k_dark.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/train/train_neurop_mit5k_dark.yaml -------------------------------------------------------------------------------- /codes/configs/train/train_neurop_mit5k_lite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/train/train_neurop_mit5k_lite.yaml -------------------------------------------------------------------------------- /codes/configs/train/train_neurop_ppr10k_a.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/train/train_neurop_ppr10k_a.yaml -------------------------------------------------------------------------------- /codes/configs/train/train_neurop_ppr10k_b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/train/train_neurop_ppr10k_b.yaml -------------------------------------------------------------------------------- /codes/configs/train/train_neurop_ppr10k_c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/configs/train/train_neurop_ppr10k_c.yaml -------------------------------------------------------------------------------- /codes/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/data/__init__.py -------------------------------------------------------------------------------- /codes/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/data/dataset.py -------------------------------------------------------------------------------- /codes/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/models/__init__.py -------------------------------------------------------------------------------- /codes/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/models/loss.py -------------------------------------------------------------------------------- /codes/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/models/model.py -------------------------------------------------------------------------------- /codes/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/models/networks.py -------------------------------------------------------------------------------- /codes/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/test.py -------------------------------------------------------------------------------- /codes/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/train.py -------------------------------------------------------------------------------- /codes/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/codes/utils.py -------------------------------------------------------------------------------- /pretrained_models/neurop_fivek_dark.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/pretrained_models/neurop_fivek_dark.pth -------------------------------------------------------------------------------- /pretrained_models/neurop_fivek_lite.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/pretrained_models/neurop_fivek_lite.pth -------------------------------------------------------------------------------- /pretrained_models/neurop_ppr_a.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/pretrained_models/neurop_ppr_a.pth -------------------------------------------------------------------------------- /pretrained_models/neurop_ppr_b.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/pretrained_models/neurop_ppr_b.pth -------------------------------------------------------------------------------- /pretrained_models/neurop_ppr_c.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/pretrained_models/neurop_ppr_c.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amberwangyili/neurop-pytorch/HEAD/requirements.txt --------------------------------------------------------------------------------