pytorch_semantic_human_matting
pytorch_semantic_human_matting copied to clipboard
I feel it is slow when training.
When I train the TNet, I feel frustrated because the train speed is too slow. I use 'aisegmentcom-matting-human-datasets' to train this model (the number of images is about 34000), batch_size=8, nearly 4400 batches per one epoch. Totally 300 epochs may train about 15 days. Is it normal? Or actually 300 epochs is no necessary?
Single GPU using 1080Ti。
what's the resolution of your input? usually the model tends to converge after 50 epochs
I use 800x600 HxW size of image to train the model's T-Net stage, but the trimap seems not too good. Is it normal?