├── LICENSE ├── README.md ├── experiments ├── cls_hrnet_w18_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w18_small_v1_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w18_small_v2_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w30_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w32_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w40_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w44_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── cls_hrnet_w48_sgd_lr5e-2_wd1e-4_bs32_x100.yaml └── cls_hrnet_w64_sgd_lr5e-2_wd1e-4_bs32_x100.yaml ├── figures ├── cls-head.png ├── cls-hrnet.png └── hrnet.png ├── lib ├── config │ ├── __init__.py │ ├── default.py │ └── models.py ├── core │ ├── evaluate.py │ └── function.py ├── models │ ├── __init__.py │ └── cls_hrnet.py └── utils │ ├── modelsummary.py │ └── utils.py ├── requirements.txt └── tools ├── _init_paths.py ├── train.py └── valid.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/README.md -------------------------------------------------------------------------------- /experiments/cls_hrnet_w18_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w18_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w18_small_v1_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w18_small_v1_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w18_small_v2_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w18_small_v2_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w30_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w30_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w32_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w32_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w40_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w40_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w44_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w44_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w48_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w48_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /experiments/cls_hrnet_w64_sgd_lr5e-2_wd1e-4_bs32_x100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/experiments/cls_hrnet_w64_sgd_lr5e-2_wd1e-4_bs32_x100.yaml -------------------------------------------------------------------------------- /figures/cls-head.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/figures/cls-head.png -------------------------------------------------------------------------------- /figures/cls-hrnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/figures/cls-hrnet.png -------------------------------------------------------------------------------- /figures/hrnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/figures/hrnet.png -------------------------------------------------------------------------------- /lib/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/config/__init__.py -------------------------------------------------------------------------------- /lib/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/config/default.py -------------------------------------------------------------------------------- /lib/config/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/config/models.py -------------------------------------------------------------------------------- /lib/core/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/core/evaluate.py -------------------------------------------------------------------------------- /lib/core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/core/function.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/cls_hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/models/cls_hrnet.py -------------------------------------------------------------------------------- /lib/utils/modelsummary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/utils/modelsummary.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/valid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HRNet/HRNet-Image-Classification/HEAD/tools/valid.py --------------------------------------------------------------------------------