PAIR-code/lit

Would LiT support Object Detection Computer Vision models?

lhphanto opened this issue · 1 comments

Question as in the title. Thanks! :-)

I think it would be possible. Image input should work fine, but you may need to do some custom UI work (https://github.com/PAIR-code/lit/wiki/frontend_development.md#custom-client--modules) to show the bounding boxes. If you wanted to just show the outputs as a list though, like [("Car", 0.85), ("Bicycle", 0.78), ...] you could use the existing SparseMultilabel() output type.