search

thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.07k forks source link

Can we use the pb file for tensorflowJS import #1053

Open espinarob opened 5 years ago

espinarob commented 5 years ago

I have a trained model, with the .pb and .meta files, is it possible to use it on the web technology especially the tensorflowjs framework? Feel free to correct if this issue is irrelevant.

MITsVision commented 4 years ago

I think you should convert your model to tensorflowJS first. use tensorflow/tfjs converter to convert your .pb model. you can follow tensorflowJs documentation as well.

command: tensorflowjs_converter --input_format tf_frozen_model --output_node_names output tiny-yolo-voc-7c.pb ./