mae_segmentation
mae_segmentation copied to clipboard
implus
reproduction of semantic segmentation using masked autoencoder (mae)
Thanks for your effort in sharing this excellent work. Can you please provide a demo of applying the pre-trained models to custom images? Does the network apply masking during training...
Hi @implus, thanks for the nice work of reproducing the segmentation results of MAE! I checked the log you provided, and noticed that ```unexpected keys``` equals to ```norm.weight, norm.bias``` https://github.com/implus/mae_segmentation/blob/main/log/20220131_012835.log#L229...
Dear, Thanks for your great work! With your offered code and hyper-parameters, I get the results as follows: ``` 2022-10-01 04:03:09,229 - mmseg - INFO - Iter(val) [16000] mIoU: 0.3869,...
你好，想问下项目代码运行的torch版本是多少呢？ 我使用torch==1.10.0，有以下的错误，把continuous操作修改为clone以及relu修改为inplace=False，还是会报相同的问题(训练的过程中，我设置了use_fp16=False) ``` RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [2, 768, 32, 32]], which is output 0 of ReluBackward0,...
Hi, thank you for the fantastic work. Can you please provide the code which runs on a single GPU? Thank you
