README.md

SHNets

stacked hourglass nets in pytorch

Commands

• Training

python train_classification.py --train-image-path ../FloorplanTransformation/pytorch/img_label_arr_train/ \
                               --val-image-path ../FloorplanTransformation/pytorch/img_label_arr_test/

python train_regression.py --augment --train-image-path ./data/ny_hei_train.txt \
                           --val-image-path ./data/ny_hei_val.txt

Papers

• Stacked Hourglass Networks for Human Pose Estimation, ECCV 2016
• Stacked U-Nets: A No-Frills Approach to Natural Image Segmentation, 2018
• Stacked U-Nets for Ground Material Segmentation in Remote Sensing Imagery, CVPRW, 2018