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

multistage loss #6

Open
soans1994 opened this issue Nov 29, 2021 · 3 comments
Open

multistage loss #6

soans1994 opened this issue Nov 29, 2021 · 3 comments

Comments

@soans1994
Copy link

soans1994 commented Nov 29, 2021

hello author,

i have some queries regarding your work. I have worked recently with multistage model. for example, I used 5 stages, 5 outputs and 5 individual losses. (So, i had to use generate 5 similar labels to compute loss.),(i think even if i dont return multiple labels, the loss is computed from the single lable returned from the dataloader.)

for stacked hourglass, should i use similar method?
edit: im training your hourglass model with mse loss, how many stages are better? also how can i choose features 128 or 256? whats the difference? more number of filters?

thank you

@david8862
Copy link
Owner

david8862 commented Nov 30, 2021

@soans1994

  1. Yes, and currently the data generator has provided multi labels as a list for train.
  2. Original paper use stack=8 and features=256, but generally stack=2 has been enough to get a reasonable result

@soans1994
Copy link
Author

thank you,

  1. Is it okay if i dont return list, but a single label?
  2. okay i will try these values.

I have updampled the stage output with 4x4 kernel to get 96x96 output since my input image is very less 96x96 image. Is it okay if i upscale the stage outputs to visualise while inferencing.

@david8862
Copy link
Owner

thank you,

  1. Is it okay if i dont return list, but a single label?

should be no since keras data generator doesn't support auto duplicate label in loss

  1. okay i will try these values.

I have updampled the stage output with 4x4 kernel to get 96x96 output since my input image is very less 96x96 image. Is it okay if i upscale the stage outputs to visualise while inferencing.

I've not tried that and not sure about the model performance

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