Home

Awesome

Max-Pooling Loss

Loss Max-Pooling for Semantic Image Segmentation

Installation

Requirements

To install PyTorch, please refer to https://github.com/pytorch/pytorch#installation.

Compiling

Some parts of Max-Pooling Loss have a native C++ implementation, which must be compiled with the following commands:

cd mpl
python build.py

Using

import mpl
import torch

max_pooling_loss = mpl.MaxPoolingLoss(ratio=0.3, p=1.7, reduce=True)
loss = torch.Tensor(1, 3, 3, 3).uniform_(0, 1)
loss = max_pooling_loss(loss)