xingyizhou / pytorch-pose-hg-3d

PyTorch implementation for 3D human pose estimation
GNU General Public License v3.0
615 stars 141 forks source link

preprcossing code in mpii.py #37

Closed chenynCV closed 5 years ago

chenynCV commented 6 years ago

Maybe the range for color augmentation is wrong?

inp[0] = np.clip(inp[0] * (np.random.random() * (0.4) + 0.6), 0, 1)
inp[1] = np.clip(inp[1] * (np.random.random() * (0.4) + 0.6), 0, 1)
inp[2] = np.clip(inp[2] * (np.random.random() * (0.4) + 0.6), 0, 1)
xingyizhou commented 5 years ago

Hi, Thanks for the note. Yes, it is wrong... However, I found using color augmentation does not affect the results...