GuoLanqing / ShadowFormer

ShadowFormer (AAAI2023), Pytorch implementation
MIT License
129 stars 17 forks source link

Question about the used dataset #7

Closed wmj183363206 closed 1 year ago

wmj183363206 commented 1 year ago

Can the ShadowFormer process the dataset without gt_mask ?

GuoLanqing commented 1 year ago

You can use an existing shadow detector to pre-predicted a shadow mask, then input it into the ShadowFormer. Or you can re-train the model without the shadow mask.

wmj183363206 commented 1 year ago

OK. Thank you for your reply.

wmj183363206 commented 1 year ago

你好,我想问一下如果我是用宽度为960高度为720的图片,这个训练指令可以是这样的吗? python train.py --warmup --win_size 12 --train_ps 720

GuoLanqing commented 1 year ago

The train_ps should be a multiple of the win_size * 8. Thus you can train with the win_size=10. Alternatively, the train_ps=720 is very large, leading to a large GPU memory. You can choose a smaller train_ps.

wmj183363206 commented 1 year ago

Thank you for your patient reply. ^ . ^

hnrna commented 1 year ago

The train_ps should be a multiple of the win_size * 8. Thus you can train with the win_size=10. Alternatively, the train_ps=720 is very large, leading to a large GPU memory. You can choose a smaller train_ps.

Have you experimented with the effect of adjusting the parameters --win_size and --train_ps on the effect of the model?

In addition does the program limit the upper limit of the parameters --win_size and --train_ps?