Computer vision models on PyTorch - View it on GitHub