xinntao / SFTGAN

CVPR18 - Recovering Realistic Texture in Image Super-resolution by Deep Spatial Feature Transform

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

generating segmentation probability mapping

charlesyou999648 opened this issue · comments

Hi Xintao, I am wondering whether it is possible for you to release the code about segmentation probability mapping.

Hi, @charlesyou999648

We use Xiaoxiao Li's codes to train our segmentation model and transfer it to a PyTorch model.

It uses a modified Caffe and may take some time to compile and solve the enviroment issues. But it has not updated for a while.

Maybe try some pytorch repo for segmentation, like semantic-segmentation-pytorch. It uses Synchronized Batch Normalization and has many pre-trained models.

I am afraid that I do not have enough time to release the training codes for segmentation in the coming several months.

BTW, a relevant issue in BasicSR.

Thanks a lot!