-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
3 Stage training
Train in parallel gt, sl, occ
- implement saving and loading of individual unet decoder branches
- When using SSIM for loss: 1 - SSIM_value (to avoid negative loss, because we want to maximise SSIM)
- implement losses
- GT:
- MSE(gt_original,
gt_prediction)
- MSE(gt_original,
- SL:
- MAE(light_generated,
light_prediction) - MAE(shadow_generated,
shadow_prediction)
- MAE(light_generated,
- OCC:
- 1 - SSIM(gt_original * shadow_generated + light_generated XOR
occ_predicted, input_image)
- 1 - SSIM(gt_original * shadow_generated + light_generated XOR
- GT:
- prepare configs
- GT:
- MSE
- SL:
- MAE
- OCC:
- SSIM
- GT:
Metadata
Metadata
Assignees
Labels
No labels