dbolya / yolact

A simple, fully convolutional model for real-time instance segmentation.
MIT License
5.02k stars 1.32k forks source link

get object coordinates #309

Open tarjoadi opened 4 years ago

tarjoadi commented 4 years ago

Hi,

Is there a way to get the coordinates if an certain object is detected ?

Thanks

bayraktare commented 4 years ago

Hi, You mean bounding box coordinates? If so, you can get them from eval.py https://github.com/dbolya/yolact/blob/db81124874817895db69f2dc443f5c24e0e3f491/eval.py#L160 in addition to the classes and confidence rates.

dbolya commented 4 years ago

Also #256 might be of use.

Marwen-Bhj commented 4 years ago

Hi, You mean bounding box coordinates? If so, you can get them from eval.py

https://github.com/dbolya/yolact/blob/db81124874817895db69f2dc443f5c24e0e3f491/eval.py#L160

in addition to the classes and confidence rates.

x what about the masks themselves I need them in order to extract the detected objects without a background

areebsyed commented 4 years ago

@Marwen-Bhj did you get any solution for masks? I also want to do that

Marwen-Bhj commented 4 years ago

@areebsyed yes you can refer to this https://github.com/dbolya/yolact/issues/286