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

1. Clone the repository
2. Install dependecines

Resources preparation

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

Training

Testing

Citation