├── .DS_Store ├── Figs ├── .DS_Store ├── NYU_res.png ├── cityscape_log.pdf ├── cityscape_log.png ├── cityscape_res.png ├── density_plot.pdf ├── density_plot.png ├── flowchart_resize.pdf ├── flowchart_resize.png └── taskonomy_res.png ├── README.md ├── cityscapes_iou.txt ├── config_cityscapes.py ├── config_nyuv2.py ├── config_taskonomy.py ├── create_pruned_net.py ├── dataloaders ├── __init__.py ├── cityscapes_dataloader.py ├── nyu_v2_dataloader.py └── taskonomy_dataloader.py ├── dynamic_prune.py ├── eval_pruned_net.py ├── evaluation.py ├── iou_analysis.py ├── iou_visualize.ipynb ├── launch_training.py ├── logs └── .DS_Store ├── loss.py ├── nyuv2_iou.txt ├── prune_utils.py ├── scene_net.py ├── scene_net_base_blocks.py ├── taskonomy_iou.txt ├── train.py └── train_basenet.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/.DS_Store -------------------------------------------------------------------------------- /Figs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/.DS_Store -------------------------------------------------------------------------------- /Figs/NYU_res.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/NYU_res.png -------------------------------------------------------------------------------- /Figs/cityscape_log.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/cityscape_log.pdf -------------------------------------------------------------------------------- /Figs/cityscape_log.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/cityscape_log.png -------------------------------------------------------------------------------- /Figs/cityscape_res.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/cityscape_res.png -------------------------------------------------------------------------------- /Figs/density_plot.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/density_plot.pdf -------------------------------------------------------------------------------- /Figs/density_plot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/density_plot.png -------------------------------------------------------------------------------- /Figs/flowchart_resize.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/flowchart_resize.pdf -------------------------------------------------------------------------------- /Figs/flowchart_resize.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/flowchart_resize.png -------------------------------------------------------------------------------- /Figs/taskonomy_res.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/Figs/taskonomy_res.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/README.md -------------------------------------------------------------------------------- /cityscapes_iou.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/cityscapes_iou.txt -------------------------------------------------------------------------------- /config_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/config_cityscapes.py -------------------------------------------------------------------------------- /config_nyuv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/config_nyuv2.py -------------------------------------------------------------------------------- /config_taskonomy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/config_taskonomy.py -------------------------------------------------------------------------------- /create_pruned_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/create_pruned_net.py -------------------------------------------------------------------------------- /dataloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/dataloaders/__init__.py -------------------------------------------------------------------------------- /dataloaders/cityscapes_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/dataloaders/cityscapes_dataloader.py -------------------------------------------------------------------------------- /dataloaders/nyu_v2_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/dataloaders/nyu_v2_dataloader.py -------------------------------------------------------------------------------- /dataloaders/taskonomy_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/dataloaders/taskonomy_dataloader.py -------------------------------------------------------------------------------- /dynamic_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/dynamic_prune.py -------------------------------------------------------------------------------- /eval_pruned_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/eval_pruned_net.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/evaluation.py -------------------------------------------------------------------------------- /iou_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/iou_analysis.py -------------------------------------------------------------------------------- /iou_visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/iou_visualize.ipynb -------------------------------------------------------------------------------- /launch_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/launch_training.py -------------------------------------------------------------------------------- /logs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/logs/.DS_Store -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/loss.py -------------------------------------------------------------------------------- /nyuv2_iou.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/nyuv2_iou.txt -------------------------------------------------------------------------------- /prune_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/prune_utils.py -------------------------------------------------------------------------------- /scene_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/scene_net.py -------------------------------------------------------------------------------- /scene_net_base_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/scene_net_base_blocks.py -------------------------------------------------------------------------------- /taskonomy_iou.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/taskonomy_iou.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/train.py -------------------------------------------------------------------------------- /train_basenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/DiSparse-Multitask-Model-Compression/HEAD/train_basenet.py --------------------------------------------------------------------------------