Please download and extract your dataset to ./data, e.g. data/StanfordDogs
Dataset list:
- data/StanfordDogs: http://vision.stanford.edu/aditya86/ImageNetDogs/
- data/StanfordCars: https://ai.stanford.edu/~jkrause/cars/car_dataset.html
- data/CUB200: http://www.vision.caltech.edu/visipedia/CUB-200.html
- data/FGVC-Aircraft: http://www.robots.ox.ac.uk/~vgg/data/fgvc-aircraft/
- data/NYUv2: https://github.com/VainF/nyuv2-python-toolkit
Visualize training with Tensorboard
tensorboard --logdir=run
python finegraind.py --dataset stanford_dogs
python nyuv2_seg.py
python nyuv2_depth.py
cd knowledge_amalgamation
python joint_scene_parsing_task_branching.py --seg_ckpt PATH_TO_YOUT_SEG_MODEL --depth_ckpt PATH_TO_YOUT_DEPTH_MODEL
python comprehensive_classification_common_feature_learning.py --car_ckpt PATH_TO_YOUR_CAR_CLASSIFIER --aircraft_ckpt PATH_TO_YOUT_AIRCRAFT_CLASSIFIER
python comprehensive_classification_layerwise_amalgamation.py --car_ckpt PATH_TO_YOUR_CAR_CLASSIFIER --aircraft_ckpt PATH_TO_YOUT_AIRCRAFT_CLASSIFIER