Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

problem when augmenting the heatmaps. #18

Open
piperod opened this issue Feb 5, 2018 · 3 comments
Open

problem when augmenting the heatmaps. #18

piperod opened this issue Feb 5, 2018 · 3 comments

Comments

@piperod
Copy link

piperod commented Feb 5, 2018

I'm having an issue with the heatmap. When I use the tester save, it saves images with the heatmap. But when I am using the client, that dimension is always 0. Its like there is not heatmap created. The paf is showing well. I am using less keypoints, not sure if this is what is messing the thing. Any help, very welcome.

@anatolix
Copy link
Owner

could you explain which tensor dimension and in which place you are talking about?

@piperod
Copy link
Author

piperod commented Feb 15, 2018

I am trying to train based on only 5 parts. I modified py_rmpe_config.py in the following way:
Left left and right parts empty as I don't looking for symmetric parts that could change label when flipped (lines 5-9) . Then the parts used (line 19) and Then I have 4 limbs that are the connections between the parts specified (lines 28-29). Also modified the coco config parts, I commented the lines for neck as I am not using neck. I also modified the generate_hdf5.py so lines (96-114) don't filter my annotations as I am using different number of keypoints.

Then when I started the notebook for inspect dataset, and start the client server, I get these dimensions out of the prints :

(1, 368, 368, 3)
(1, 46, 46, 14)
(1, 46, 46, 7)
(1, 46, 46, 14)
(1, 46, 46, 0)
(1, 46, 46, 14)
(1, 46, 46, 0)
(1, 46, 46, 14)
(1, 46, 46, 0)
(1, 46, 46, 14)
(1, 46, 46, 0)
(1, 46, 46, 14)
(1, 46, 46, 0)
(1, 46, 46, 14)
(1, 46, 46, 0)

For some reason the dimensions for the heatmaps are 0 for channels, so I am not receiving the annotations.

@anatolix
Copy link
Owner

  1. could you copy/link config somewhere it is really hard to find problem
  2. config could be run as python program, in this case it prints layer map.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants