search

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 400 forks source link

Doubts regarding the predict module #250

Open AshwinAce opened 5 years ago

AshwinAce commented 5 years ago

When we run lumi predict is it possible to display the bounding boxes around an image? Is it possible to get the locations and probabilities of the objects stored in say an excel file with the name of the images stored as well?

Is there a way to save the predictions inside a list, for instance?

lartpang commented 5 years ago

image

I didn't find a way to display the results, too. maybe the code here doesn't do that.

https://github.com/tryolabs/luminoth/blob/ad27c535576d544adc868b5bb9ce2269bf04c22c/luminoth/predict.py#L174

and about the way to store the data, the -f option maybe can help you. like this..

lumi predict 1131868480.jpg -f predict.json

About what formats it supports, you can look these codes about it.

https://github.com/tryolabs/luminoth/blob/ad27c535576d544adc868b5bb9ce2269bf04c22c/luminoth/predict.py#L223

I hope it'll help you.

AshwinAce commented 5 years ago

This is helpful. What would happen when we run predict on multiple images. Would they all be saved in a single json? The comments seem to indicate that although I'm not sure whether the names would be saved as well.

lartpang commented 5 years ago

https://github.com/tryolabs/luminoth/blob/ad27c535576d544adc868b5bb9ce2269bf04c22c/luminoth/predict.py#L210

https://github.com/tryolabs/luminoth/blob/ad27c535576d544adc868b5bb9ce2269bf04c22c/luminoth/predict.py#L265

you can look this code. maybe it can do it...

Take a moment to look at the code and you'll probably know what it can and can't do.

I hope it'll help you.

lartpang commented 5 years ago

image

image