TensorFlow implementation of Box2Seg: Attention Weighted Loss and Discriminative Feature Learning for Weakly Supervised Segmentation.