Train with a customized image background

[MatteoMarengo]

Hello !

Do you have any advice on which part of the code should I modify in order to do the training with a customized image as a background instead of using a background color ?

Thank you so much :)

[liruilong940607]

Hi,

That is fairly easy! Here is the pseudo code.

backgrounds: ... # [C, H, W, 1]
# colors is [C, H, W, 3], alphas is [C, H, W, 1]
colors, alphas, meta = gsplat.rasterization(...)
# apply background
colors = colors + backgrounds * (1 - alphas)

[MatteoMarengo]

In which files should I apply these modifications ?

Thank you ! :)

[liruilong940607]

simple_trainer.py