search

huochaitiantang / pytorch-deep-image-matting

Pytorch implementation of deep image matting
294 stars 71 forks source link

absence of normalization #12

Open whyjay opened 5 years ago

whyjay commented 5 years ago

Hi, thanks for great code. I inspected the input stream and found that input image is not normalized. As far as i know, pytorch VGG assumes normalized pixel values. RGB value should be divided by 255, then normalized with mean=[0.485, 0.456, 0.406] and std=[0.229, 0.224, 0.225].

huochaitiantang commented 5 years ago

Thanks for your review! The issue may lead to bad performance. I will conduct experiments with normalized input later.