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

Training the segmentation code #11

Closed
halbielee opened this issue Jul 19, 2020 · 11 comments
Closed

Training the segmentation code #11

halbielee opened this issue Jul 19, 2020 · 11 comments

Comments

@halbielee
Copy link

Hello!

Thank you for sharing the excellent code.

I am trying to reproduce the performance you reported and I tried to train the result of the affinity network [Ahn et al.] with the segmentation code of https://github.com/itijyou/ademxapp

But I failed to train. Can you share the hyper-parameters or any change when you train?
From the affinity net I found that he changed SGD to Adam with his work.

You may not remember, I need a little clue.

Thank you.

@YudeWang
Copy link
Owner

Hi @halbielee ,

The original segmentaion code is not available for me at present because the server is down and not fixed yet. I have a copy of the project but the hyper-parameters have been changed few times and I am not very clear about the exactly setting for the paper experiments. I will list the present parameters in that configuration file.

The segmentation model is Deeplabv1 with resnet38 backbone (maybe you need to add some Deeplab module on resnet38 from the link you given). Noting that the additional layers' learning rate is 10x than backbone layers.
SGD with lr=0.001, weight_decay=0.0005, poly policy with 0.9,
GPUS=4, batchsize=10, itr=20k,
random crop_size=321 (maybe 448),
random rescale [0.5-2.0],
random flip,
random color jitter
multi-scale test [0.5, 0.75, 1.0, 1.25, 1.5, 1.75]

@halbielee
Copy link
Author

Thank you @YudeWang

I will try that and let you know the result.!

@TyroneLi
Copy link

Thank you @YudeWang

I will try that and let you know the result.!

Hi~have you solved this problem? Which deeplab code repro you chosed finally? could you share a link or details with me? Thansk

@halbielee
Copy link
Author

Hello @TyroneLi
I could not solve this. When I modify this with DeepLab module, it shows 3-5 % lower performance than the normal WideResNet38 network.

@YudeWang
Copy link
Owner

@TyroneLi @halbielee
Here is a screenshot of deeplabv1 file, maybe it will help you.
image

@Gzn520
Copy link

Gzn520 commented Oct 30, 2020

请问,在推断时,cam 和crf 是训练时得到的吗 为什么 没有看到生成呀?

@Gzn520
Copy link

Gzn520 commented Nov 3, 2020

如何让cam可视化?

@zhudahui
Copy link

@Gzn520 请问您解决了吗

@zbf1991
Copy link

zbf1991 commented Dec 15, 2020

@YudeWang I try to re-implement deeplab-resnet38 following your screenshot, but the performance is much lower, could you please share your segmentation code?

@YudeWang
Copy link
Owner

YudeWang commented Jan 8, 2021

@TyroneLi @halbielee @zbf1991 @zhudahui
Retrain step code is released here with approximate performance comparing to paper.

@YudeWang
Copy link
Owner

YudeWang commented Jan 8, 2021

Any question about retrain step can be given in new repository and I will close this issue.

@YudeWang YudeWang closed this as completed Jan 8, 2021
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

6 participants