Part Features

[asafjo23]

Hi Vibe, Great Work!

I wanted to ask, I need the part features before the heat maps.
Altho, from the code I see that the final step of the part features is the _get_part_attention_map function.
Afterwards, I see that there is an if statement that says: elif self.use_heatmaps == 'part_segm' then output['pred_segm_mask'] = heatmaps.
I wanted to ask, in this case the pred_segm_mask are the body part segments as you attached in the appendix of the paper?
(see attached picture :-) )