search

carpentries-incubator / intro-image-classification-cnn

new lesson on image classification with convolutional neural networks
https://carpentries-incubator.github.io/intro-image-classification-cnn/
Other
2 stars 3 forks source link

discuss how to take model from one library and run in another #40

Open erinmgraham opened 4 months ago

erinmgraham commented 4 months ago

can .keras models be used in eg pytorch? explain or refer to where to look

IsaacJennings00 commented 3 months ago

To go from Keras to PyTorch, models must be converted to an open format called ONNX (https://onnx.ai/) Using tensorflow-onnx first (https://github.com/onnx/tensorflow-onnx) followed by onnx2pytorch (https://github.com/Talmaj/onnx2pytorch)