Various neural network models for PyTorch.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Failed to load latest commit information.


Various neural network models for PyTorch. Weights were converted from Caffe models using my fork of caffe-tensorflow.


Weights are stored in an order consistent with PyTorch. Refer to kitnn.utils to see how to read them in. They can be downloaded here:

Material in Context

Bell, Sean, et al. "Material recognition in the wild with the materials in context database." Proceedings of the IEEE conference on computer vision and pattern recognition. 2015.
  • MINC AlexNet
  • MINC VGG-16
  • MINC GoogLeNet


  • VGG-19 (normalized)