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

Name		Name	Last commit message	Last commit date
Latest commit History 16 Commits
models		models
LICENSE		LICENSE
README.md		README.md
check_result_classification.ipynb		check_result_classification.ipynb
check_result_regression.ipynb		check_result_regression.ipynb
dataloader.py		dataloader.py
dataloader_classifier.py		dataloader_classifier.py
dataloader_regressor.py		dataloader_regressor.py
image_augment.py		image_augment.py
image_augment_pairs.py		image_augment_pairs.py
loss.py		loss.py
train.py		train.py
train_classification.py		train_classification.py
train_regression.py		train_regression.py
utils.py		utils.py

License

shuuchen/SHNets

Folders and files

Latest commit

History

Repository files navigation

SHNets

Commands

Papers

About

Topics

Resources

License

Stars

Watchers

Forks

Languages