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.59k stars 2.19k forks source link

How to output the custom predict result? #152

Closed woshichuanqilz closed 5 years ago

woshichuanqilz commented 5 years ago

I want the output image with mark rectangle on the image.

I check the function

def predictImage(self, image_input, result_count=1, input_type="file"):

seems no output option.

How to make it?

OlafenwaMoses commented 5 years ago

You can only get the box points to create a rectangular mark on the object in an image if you are using an object detection model. ImageAI doesn't support detecting custom objects yet. Keep up to date with #8 to know the latest on this.