├── .gitignore ├── LICENSE ├── README.md ├── asserts ├── overview.png └── result_in_paper.png ├── lib ├── __init__.py ├── config.py ├── datasets │ └── lm.py ├── model.py ├── models │ ├── CDPN.py │ ├── __init__.py │ ├── resnet_backbone.py │ ├── resnet_rot_head.py │ └── resnet_trans_head.py ├── ref.py ├── test.py ├── train.py └── utils │ ├── __init__.py │ ├── eval.py │ ├── fancy_logger.py │ ├── fs.py │ ├── img.py │ ├── io.py │ ├── tictoc.py │ ├── transform3d.py │ └── utils.py └── tools ├── _init_paths.py ├── exps_cfg ├── config_rot.yaml ├── config_rot_trans.yaml └── config_trans.yaml ├── main.py ├── train_stage1.sh ├── train_stage2.sh └── train_stage3.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/README.md -------------------------------------------------------------------------------- /asserts/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/asserts/overview.png -------------------------------------------------------------------------------- /asserts/result_in_paper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/asserts/result_in_paper.png -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/__init__.py -------------------------------------------------------------------------------- /lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/config.py -------------------------------------------------------------------------------- /lib/datasets/lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/datasets/lm.py -------------------------------------------------------------------------------- /lib/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/model.py -------------------------------------------------------------------------------- /lib/models/CDPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/models/CDPN.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/models/resnet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/models/resnet_backbone.py -------------------------------------------------------------------------------- /lib/models/resnet_rot_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/models/resnet_rot_head.py -------------------------------------------------------------------------------- /lib/models/resnet_trans_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/models/resnet_trans_head.py -------------------------------------------------------------------------------- /lib/ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/ref.py -------------------------------------------------------------------------------- /lib/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/test.py -------------------------------------------------------------------------------- /lib/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/train.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/eval.py -------------------------------------------------------------------------------- /lib/utils/fancy_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/fancy_logger.py -------------------------------------------------------------------------------- /lib/utils/fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/fs.py -------------------------------------------------------------------------------- /lib/utils/img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/img.py -------------------------------------------------------------------------------- /lib/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/io.py -------------------------------------------------------------------------------- /lib/utils/tictoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/tictoc.py -------------------------------------------------------------------------------- /lib/utils/transform3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/transform3d.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/exps_cfg/config_rot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/tools/exps_cfg/config_rot.yaml -------------------------------------------------------------------------------- /tools/exps_cfg/config_rot_trans.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/tools/exps_cfg/config_rot_trans.yaml -------------------------------------------------------------------------------- /tools/exps_cfg/config_trans.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/tools/exps_cfg/config_trans.yaml -------------------------------------------------------------------------------- /tools/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/tools/main.py -------------------------------------------------------------------------------- /tools/train_stage1.sh: -------------------------------------------------------------------------------- 1 | export QT_QPA_PLATFORM='offscreen' 2 | ulimit -n 10240 3 | python main.py --cfg=exps_cfg/config_rot.yaml 4 | 5 | -------------------------------------------------------------------------------- /tools/train_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LZGMatrix/CDPN_ICCV2019_ZhigangLi/HEAD/tools/train_stage2.sh -------------------------------------------------------------------------------- /tools/train_stage3.sh: -------------------------------------------------------------------------------- 1 | export QT_QPA_PLATFORM='offscreen' 2 | ulimit -n 10240 3 | python main.py --cfg=exps_cfg/config_rot_trans.yaml 4 | 5 | --------------------------------------------------------------------------------