search

PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
https://qiita.com/PINTO
MIT License
3.62k stars 574 forks source link

Tfjs Modnet model gives strange result #75

Closed mtuan4i closed 3 years ago

mtuan4i commented 3 years ago

First, thank you very much for your great work. I tried inference with tfjs Modnet model but the result is not as expected. My inference code is based on https://github.com/PINTO0309/PINTO_model_zoo/blob/main/078_MODNet/test_tflite.py Here is what I got. image

This result is same as your result on tflite?

PINTO0309 commented 3 years ago

I have re-committed the model. Please try again. https://drive.google.com/file/d/1Y6B6l72hib9t5ammT-7ya-muBzRKZ7xq/view?usp=sharing Screenshot 2021-02-28 01:11:22

I used the following tool to switch to fully automatic conversion to avoid mistakes. ONNX -> OpenVINO -> tflite / tfjs https://github.com/PINTO0309/openvino2tensorflow.git a

mtuan4i commented 3 years ago

It works now. Thank you for the new model and the conversion tool

msurguy commented 3 years ago

@mtuan4i could you provide a sample code of how to use this model in TFjs? I'm beginning learning about this and it would be great to see an example. Thanks!