CODEWITHSUNDEEP
machine-learningneural-networktensorflowconv-neural-network
trainset
trainset

Reputation: 2249

Tensorflow model zoo?

One of the main advantages of caffe for me was the possibility of doing transfer learning on freely distributed pretrained models.

Is there a place to get trained models from papers/competitions in tensorflow format?

If not, is there a possibility to convert existing caffe(or any other) models into tensorflow models?

Upvotes: 15

Views: 9884

Answers (4)

tensor
tensor

Reputation: 3360

There are different ways to do Transfer learning in TFlow. 1st solution is to write your save/load on checkpoints. However, there are incompatibilities with TF 1.0 and TF 2.0

Other one is to used some pre-framework on top of TF, some of them are : TFLayers : TF specific: TF Slim MLModels : Cross platform model zoo: MLmodels

Upvotes: 0

deramos
deramos

Reputation: 48

There are a few ways to do transfer learning with Tensorflow some of which combine TF Slim and custom codes, but there is a very nice collection of pretrained Tensorflow models Tensornets which contain almost all popular models and their pretrained weights

Upvotes: 0

Salvador Dali
Salvador Dali

Reputation: 222999

As of now, official TF github page has a list with the group of models. Also you can use keras.applications which has pretrained models.

Upvotes: 2

dga
dga

Reputation: 21927

You can likely use the caffe to tensorflow model converter to convert model zoo models. If you try it and report back, it would be great to know. There's a potential issue with converting maxpooling and padding, but it seems to work for many models.

Upvotes: 17

Related Questions