Object Extraction

[HaQuocHuyPtit]

I have a question. How can I extract object after detecting one? Thank you.

[zhixuan-lin]

Hello,

There is no code in this repo that visualizes individual objects, but you can write your own (it is pretty easy). y_att here (of shape (B, G*G, 3, H, W)) are the extracted (actually reconstructed) objects. You may want to add it to the log here so you get it outside this function. Also, you may want to select non-empty objects from y_att with z_pres here (shape (B, G*G, 1), you will need to binarize it).