PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
https://peterl1n.github.io/RobustVideoMatting/
GNU General Public License v3.0
8.61k stars 1.14k forks source link

Data composition and augmentation #48

Closed FengMu1995 closed 3 years ago

FengMu1995 commented 3 years ago

The part "data augmentation and composition" has been added into the part "Training" ? I wonder if I should do data augmentation independent of your code?

PeterL1n commented 3 years ago

All the augmentation code are in https://github.com/PeterL1n/RobustVideoMatting/blob/master/dataset/augmentation.py

It is part of the train script and performed on-the-fly.