GitHub - Changgang-Zheng/HD-CNN: The realization of paper 'HD-CNN: Hierarchical Deep Convolutional Neural Networks for Large Scale Visual Recognition'

HD-CNN

remember to chage the data address: 'root_path' in dataset.py
Change: 'Determine which dataset to be used ================ cifar-10/cifar-100'
Change: 'The number of cluster centers ======================= 2/9'
Change: 'The number of pre-train steps ================== when final test, make it bigger'
Change: 'The number of test batch steps in a epoch ================= when final test, make it bigger'
Change: 'The number of fine classes ======================= 10/100'
Change (optional): two parameters: 'u_t= 1/(args.num_superclass*5)' in main.py and 'lam=20' in loss.py

I am not sure if cuda works
I am not sure if fine-tune part have problem as the accuracy looks a little weird I am trying to find why

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
HD-CNN-Final		HD-CNN-Final
__pycache__		__pycache__
logging/model_dir		logging/model_dir
HD-CNN for Large Scale Visual Recognition.md		HD-CNN for Large Scale Visual Recognition.md
HD-CNN, Hierarchical Deep Convolutional Neural Networks for Large Scale Visual Recognition.pdf		HD-CNN, Hierarchical Deep Convolutional Neural Networks for Large Scale Visual Recognition.pdf
config.py		config.py
dataset.py		dataset.py
losses.py		losses.py
main.py		main.py
models.py		models.py
readme.md		readme.md
spectrual_clustering.py		spectrual_clustering.py
utils.py		utils.py