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.
Permalink
Failed to load latest commit information.
kitnn
.gitignore
README.md

README.md

kitnn

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

Weights

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: http://bit.ly/2urbzGy

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

  • VGG-19 (normalized)