search

PeterL1n / BackgroundMattingV2

Real-Time High-Resolution Background Matting
MIT License
6.81k stars 950 forks source link

Details of the training process #190

Open fenneishi opened 2 years ago

fenneishi commented 2 years ago

你好，我测试了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 再次感谢！