Skip to content

Latest commit

 

History

History
30 lines (25 loc) · 2.14 KB

adding_new_model.md

File metadata and controls

30 lines (25 loc) · 2.14 KB
Raw

Adding New Models

Adding New Model - Caffe

  1. For Setup instructions, look at the README.
  2. Add the new model as a .prototxt file to the Example/Caffe folder.
  3. Then add your entry to the front-end by refering to the following example in modelZoo.js, in this the id should be the name of your prototxt without the extension.
<li><ModelElement importNet={this.props.importNet} framework="caffe" id="sample">Sample</ModelElement></li>
  1. After making these changes, test if loading the model and exporting it to both or at least one framework is working fine and document it accordingly in your pull request.
  2. Add a thumbnail image for displaying a preview of the new model.
  3. Add the new model to Tested Models.
  4. Create a pull request for the same and get reviewed by the mentors. Cheers!

Adding New Model - Keras

  1. For Setup instructions, look at the README.
  2. Add the new model as a .json file to the Example/Keras folder.
  3. Then add your entry to the front-end by refering to the following example in modelZoo.js, in this the id should be the name of your json without the extension.
<li><ModelElement importNet={this.props.importNet} framework="keras" id="Sample">sample</ModelElement></li>
  1. After making these changes, test if loading the model and exporting it to both or at least one framework is working fine and document it accordingly in your pull request.
  2. Add a thumbnail image for displaying a preview of the new model.
  3. Add the new model to Tested Models.
  4. Create a pull request for the same and get reviewed by the mentors. Cheers!