longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

Normalize inputs #37

Open lematt1991 opened 6 years ago

lematt1991 commented 6 years ago

Should we be normalizing the inputs prior to feeding them into the network [1]? All of the PyTorch networks contained in the model zoo use the same mean and std deviation for preprocessing images:

The images have to be loaded in to a range of [0, 1] and then normalized using mean=[0.485, 0.456, 0.406] and std=[0.229, 0.224, 0.225]