Semi-Supervised Wide-Angle Portraits Correction by Multi-Scale Transformer

Fushun Zhu, Shan Zhao, Peng Wang, Hao Wang, Hua Yan, Shuaicheng Liu

The pipline of semi-supervised wide-angle portraits correction framework with the surrogate task (segmentation)

Note

In this repository, we will release the unlabeled dataset and MegDL implementation of our paper.

Quick Start

All codes are tested on Linux.

Installation

Clone the repository
Install dependecines

Resources preparation

Dataset GoogleDrive BaiduCloud(Extraction code:79cw)
Pre-trained model

Training

Testing

Citation