├── checkpoint ├── baseline_msunet.tar └── semi_supervised.con_seg.tar ├── losses ├── L1_mask_Loss.py ├── __init__.py ├── landmark_loss.py ├── sobel.py └── sobel_mask_loss.py ├── models ├── __init__.py └── transformer.py ├── readme.md ├── semi-supervised framework.PNG ├── test.py ├── test.sh ├── train.py ├── train.sh └── utils ├── Calculator.py ├── __init__.py ├── dataset.py ├── mytrainer.py ├── myutils.py └── randaugment.py /checkpoint/baseline_msunet.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/checkpoint/baseline_msunet.tar -------------------------------------------------------------------------------- /checkpoint/semi_supervised.con_seg.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/checkpoint/semi_supervised.con_seg.tar -------------------------------------------------------------------------------- /losses/L1_mask_Loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/losses/L1_mask_Loss.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/landmark_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/losses/landmark_loss.py -------------------------------------------------------------------------------- /losses/sobel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/losses/sobel.py -------------------------------------------------------------------------------- /losses/sobel_mask_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/losses/sobel_mask_loss.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/models/transformer.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/readme.md -------------------------------------------------------------------------------- /semi-supervised framework.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/semi-supervised framework.PNG -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/test.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/test.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/train.sh -------------------------------------------------------------------------------- /utils/Calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/utils/Calculator.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/mytrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/utils/mytrainer.py -------------------------------------------------------------------------------- /utils/myutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/utils/myutils.py -------------------------------------------------------------------------------- /utils/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Portraits_Correction/HEAD/utils/randaugment.py --------------------------------------------------------------------------------