PeterL1n / BackgroundMattingV2

Real-Time High-Resolution Background Matting

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Details of the training process

fenneishi opened this issue · comments

你好,我测试了BackgroundMattingV2,发现效果很好,感谢。
但是关于一些训练的细节,我还是不太理解(PS:主要是关于数据增强部分)

  1. 在train_base中,针对 dataset_train的Affine增强,为什么fgr-pha与bgr在scale参数上的随机范围不一样,或者说为什么分别设置成这个范围
    image
  2. 在train_base中,通过ZipDataset将pha_fgr_dataset和bgr_dataset打包一个整体dataset(下图1),但是dataset好像是一种固定匹配的打包方式,固定匹配不是会造成合成的数据量远远小于len(fgr)*len(bgr)吗?(下图2),这样会不会有一些问题。
    image
    image
    再次感谢!