search

OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.5k stars 2.18k forks source link

Can i use model obtained from Custom object detection training in tensorflowjs? #564

Open nachiketmore99 opened 3 years ago

nachiketmore99 commented 3 years ago

@OlafenwaMoses I want help with implementing a yolov3 custom trained model to tensorflowjs.

Combinacijus commented 3 years ago

ImageAI in custom model training generates two files model.h5 and detection_config.json which by the looks of things could be imported to tensorflowjs.

It hard to help more without actually learning tensorflowjs and writing all code for you. But if you have more questions on ImageAI side feel free to ask.

If is solved the issue please close it.

OlafenwaMoses commented 3 years ago

@nachiketmore99 You might find this useful.

https://www.tensorflow.org/js/tutorials/conversion/import_keras