Adding channel to pretrained PSPNet input

Hello all,

I am performing image segmentation with PSPnet and would like to initialize the parameters of the first Conv layer (B x 3 x H x W) of the pretrained model with an extra channel of zeros (to create B x 4 x H x W).

Does anybody have recommendations on the best way to do this?

Thank you!