How can I output AE heatmap?

princeton-vl / pose-ae-train

Training code for "Associative Embedding: End-to-End Learning for Joint Detection and Grouping"

BSD 3-Clause "New" or "Revised" License

373 stars 76 forks source link

How can I output AE heatmap? #33

Open ArchNew opened 5 years ago

ArchNew commented 5 years ago

I want to output AE predictions and visualize it. Where could I get it in the code? I mean...according to your paper, you have heatmaps of tags at every pixel position. And also in the paper, you visualized those heatmaps. I want to output those heatmaps of my own picture, what should I do? Or where, in the code, can I intercept such "heatmaps of tags" (AE predictions) to output?

BoltenWang-Meta commented 2 years ago

Emm, have you done with this issue? This code is so complex to read...