├── .gitignore ├── LICENSE.md ├── README.md ├── app ├── cifar10 │ ├── build_basenet.py │ └── build_resnet.py ├── imagenet │ ├── build_alexnet.py │ ├── build_resnet.py │ ├── build_squeezenet.py │ └── build_vggnet.py └── ssd │ └── create_ssdnet.py ├── config └── default.params ├── models ├── cifar10 │ ├── Readme.md │ └── resnet_110 │ │ ├── solver.prototxt │ │ ├── test.prototxt │ │ └── train.prototxt ├── imagenet │ ├── Readme.md │ ├── docs │ │ ├── main_branch.png │ │ └── resnet.png │ ├── resnet_34_inception_trick │ │ ├── test.prototxt │ │ └── train.prototxt │ ├── resnet_36_inception_trick_with4k │ │ ├── test.prototxt │ │ └── train.prototxt │ ├── resnet_50 │ │ ├── plot_accuracy.jpg │ │ ├── plot_loss.jpg │ │ ├── test.prototxt │ │ └── train.prototxt │ ├── resnet_50_1by2 │ │ ├── plot_accuracy.jpg │ │ ├── plot_loss.jpg │ │ ├── test.prototxt │ │ └── train.prototxt │ ├── resnet_66_1by2_inception_trick_bottleneck │ │ ├── test.prototxt │ │ └── train.prototxt │ ├── resnet_68_1by2_inception_trick_bottleneck_with4k │ │ ├── test.prototxt │ │ └── train.prototxt │ └── resnet_77_1by2 │ │ ├── plot_accuracy.jpg │ │ ├── plot_loss.jpg │ │ ├── test.prototxt │ │ └── train.prototxt └── voc2007_ssd │ ├── Readme.md │ └── resnet_50 │ ├── plot_loss.jpg │ ├── solver.prototxt │ ├── solver_score.prototxt │ ├── test.prototxt │ └── train.prototxt ├── netbuilder ├── __init__.py ├── lego │ ├── __init__.py │ ├── base.py │ ├── data.py │ ├── hybrid.py │ └── ssd.py ├── nets │ ├── __init__.py │ ├── imagenet.py │ └── ssdnet.py └── tools │ ├── __init__.py │ └── complexity.py ├── setup.py └── tests ├── __init__.py ├── lego ├── __init__.py ├── test_base.py ├── test_data.py ├── test_hybrid.py └── test_ssd.py └── tools ├── __init__.py └── test_complexity.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/README.md -------------------------------------------------------------------------------- /app/cifar10/build_basenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/cifar10/build_basenet.py -------------------------------------------------------------------------------- /app/cifar10/build_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/cifar10/build_resnet.py -------------------------------------------------------------------------------- /app/imagenet/build_alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/imagenet/build_alexnet.py -------------------------------------------------------------------------------- /app/imagenet/build_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/imagenet/build_resnet.py -------------------------------------------------------------------------------- /app/imagenet/build_squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/imagenet/build_squeezenet.py -------------------------------------------------------------------------------- /app/imagenet/build_vggnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/imagenet/build_vggnet.py -------------------------------------------------------------------------------- /app/ssd/create_ssdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/app/ssd/create_ssdnet.py -------------------------------------------------------------------------------- /config/default.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/config/default.params -------------------------------------------------------------------------------- /models/cifar10/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/cifar10/Readme.md -------------------------------------------------------------------------------- /models/cifar10/resnet_110/solver.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/cifar10/resnet_110/solver.prototxt -------------------------------------------------------------------------------- /models/cifar10/resnet_110/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/cifar10/resnet_110/test.prototxt -------------------------------------------------------------------------------- /models/cifar10/resnet_110/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/cifar10/resnet_110/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/Readme.md -------------------------------------------------------------------------------- /models/imagenet/docs/main_branch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/docs/main_branch.png -------------------------------------------------------------------------------- /models/imagenet/docs/resnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/docs/resnet.png -------------------------------------------------------------------------------- /models/imagenet/resnet_34_inception_trick/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_34_inception_trick/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_34_inception_trick/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_34_inception_trick/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_36_inception_trick_with4k/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_36_inception_trick_with4k/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_36_inception_trick_with4k/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_36_inception_trick_with4k/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_50/plot_accuracy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50/plot_accuracy.jpg -------------------------------------------------------------------------------- /models/imagenet/resnet_50/plot_loss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50/plot_loss.jpg -------------------------------------------------------------------------------- /models/imagenet/resnet_50/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_50/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_50_1by2/plot_accuracy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50_1by2/plot_accuracy.jpg -------------------------------------------------------------------------------- /models/imagenet/resnet_50_1by2/plot_loss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50_1by2/plot_loss.jpg -------------------------------------------------------------------------------- /models/imagenet/resnet_50_1by2/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50_1by2/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_50_1by2/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_50_1by2/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_66_1by2_inception_trick_bottleneck/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_66_1by2_inception_trick_bottleneck/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_66_1by2_inception_trick_bottleneck/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_66_1by2_inception_trick_bottleneck/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_68_1by2_inception_trick_bottleneck_with4k/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_68_1by2_inception_trick_bottleneck_with4k/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_68_1by2_inception_trick_bottleneck_with4k/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_68_1by2_inception_trick_bottleneck_with4k/train.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_77_1by2/plot_accuracy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_77_1by2/plot_accuracy.jpg -------------------------------------------------------------------------------- /models/imagenet/resnet_77_1by2/plot_loss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_77_1by2/plot_loss.jpg -------------------------------------------------------------------------------- /models/imagenet/resnet_77_1by2/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_77_1by2/test.prototxt -------------------------------------------------------------------------------- /models/imagenet/resnet_77_1by2/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/imagenet/resnet_77_1by2/train.prototxt -------------------------------------------------------------------------------- /models/voc2007_ssd/Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/voc2007_ssd/Readme.md -------------------------------------------------------------------------------- /models/voc2007_ssd/resnet_50/plot_loss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/voc2007_ssd/resnet_50/plot_loss.jpg -------------------------------------------------------------------------------- /models/voc2007_ssd/resnet_50/solver.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/voc2007_ssd/resnet_50/solver.prototxt -------------------------------------------------------------------------------- /models/voc2007_ssd/resnet_50/solver_score.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/voc2007_ssd/resnet_50/solver_score.prototxt -------------------------------------------------------------------------------- /models/voc2007_ssd/resnet_50/test.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/voc2007_ssd/resnet_50/test.prototxt -------------------------------------------------------------------------------- /models/voc2007_ssd/resnet_50/train.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/models/voc2007_ssd/resnet_50/train.prototxt -------------------------------------------------------------------------------- /netbuilder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /netbuilder/lego/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/lego/__init__.py -------------------------------------------------------------------------------- /netbuilder/lego/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/lego/base.py -------------------------------------------------------------------------------- /netbuilder/lego/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/lego/data.py -------------------------------------------------------------------------------- /netbuilder/lego/hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/lego/hybrid.py -------------------------------------------------------------------------------- /netbuilder/lego/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/lego/ssd.py -------------------------------------------------------------------------------- /netbuilder/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/nets/__init__.py -------------------------------------------------------------------------------- /netbuilder/nets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/nets/imagenet.py -------------------------------------------------------------------------------- /netbuilder/nets/ssdnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/nets/ssdnet.py -------------------------------------------------------------------------------- /netbuilder/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/tools/__init__.py -------------------------------------------------------------------------------- /netbuilder/tools/complexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/netbuilder/tools/complexity.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/lego/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/lego/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/tests/lego/test_base.py -------------------------------------------------------------------------------- /tests/lego/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/tests/lego/test_data.py -------------------------------------------------------------------------------- /tests/lego/test_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/tests/lego/test_hybrid.py -------------------------------------------------------------------------------- /tests/lego/test_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/tests/lego/test_ssd.py -------------------------------------------------------------------------------- /tests/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tools/test_complexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jay-mahadeokar/pynetbuilder/HEAD/tests/tools/test_complexity.py --------------------------------------------------------------------------------