arnab39 / FewShot_GAN-Unet3D

Tensorflow implementation of our paper: Few-shot 3D Multi-modal Medical Image Segmentation using Generative Adversarial Learning
MIT License
377 stars 82 forks source link

How did you find class weights? #13

Open take5v opened 4 years ago

take5v commented 4 years ago

Dear Arnab,

It's not clear for me how did you choose the class weights? Did you estimate weights similar to compute_class_weight only from the training data? Did you fine-tune the weights manually?

Background- 0.33, CSF- 1.5, GM- 0.83, WM- 1.33: https://github.com/arnab39/FewShot_GAN-Unet3D/blob/master/unet3D/model_unet.py#L152.

btw Thanks a lot for sharing your code and paper. Sadly, there was nothing said about class weights in your paper.

Best regards, Stas