PyTorch code for "Locating objects without bounding boxes" - Loss function and trained models - View it on GitHub