/ModelFeast

Pytorch model zoo for human, include all kinds of 2D CNN, 3D CNN, and CRNN

Primary LanguagePythonMIT LicenseMIT

ModelFeast

中文版readme

Please star ModelFeast if it helps you. This is very important to me! Thanks very much !

ModelFeast is more than model-zoo! It is:

What is ModelFeast ?

Avalible models

2D CNN

Pretrained models ( trained on ImageNet ) for 2D CNN is now avalible on Baiduyun(fst6) and Google Drive

3D CNN

CNN-RNN

This part is still on progress. Not avalible to train now, but model architecture can been seen here.

Get started

Determine what you need and read corresponding tutorials

Or you can use modelfeast simply via pip !

pip3 install modelfeast

pip user guide

Features

The features are more than you could think of:

  • Train and save model within 3 lines !
  • Multi GPU support !
  • Include the most popular 2D CNN, 3D CNN, and CRNN models !
  • Allow any input image size (pytorch official model zoo limit your input size harshly) !
  • Help you sweep all kinds of classification competitions.

Reference

https://github.com/lanpa/tensorboardX

https://github.com/pytorch/vision/tree/master/torchvision/models

https://github.com/kenshohara/3D-ResNets-PyTorch

https://github.com/victoresque/pytorch-template

https://github.com/AlexHex7/Non-local_pytorch

https://github.com/Cadene/pretrained-models.pytorch