models

😎 Supported Models

Methods	Source	Teacher	Student
PF-AFN	Parser-Free Virtual Try-on via Distilling Appearance Flows	✅	✅
FS-VTON	Style-Based Global Appearance Flow for Virtual Try-On	✅	✅
RMGN	RMGN: A Regional Mask Guided Network for Parser-free Virtual Try-on	❌	✅
DM-VTON (Ours)	DM-VTON: Distilled Mobile Real-time Virtual Try-On	✅	✅

PF-AFN

from models.warp_modules.afwm import AFWM
from models.generators.res_unet import ResUnetGenerator

# Teacher
warp_model = AFWM(45)
gen_model  = ResUnetGenerator(8, 4, 5)

# Student
warp_model = AFWM(3)
gen_model  = ResUnetGenerator(7, 4, 5)

FS-VTON

from models.warp_modules.style_afwm import StyleAFWM
from models.generators.res_unet import ResUnetGenerator

# Teacher
warp_model = StyleAFWM(45)
gen_model  = ResUnetGenerator(8, 4, 5)

# Student
warp_model = StyleAFWM(3)
gen_model  = ResUnetGenerator(7, 4, 5)

RMGN

from models.warp_modules.afwm import AFWM
from models.generators.rmgn_generator import RMGNGenerator

# Student
warp_model = AFWM(3)
gen_model  = RMGNGenerator(in_person_nc=3, in_clothes_nc=4)

Note: RMGN only release inference code for the Student network.

DM-VTON

from models.warp_modules.mobile_afwm import MobileAFWM
from models.generators.mobile_unet import MobileNetV2_unet

# Teacher
warp_model = MobileAFWM(45)
gen_model  = MobileNetV2_unet(8, 4)

# Student
warp_model = MobileAFWM(3)
gen_model  = MobileNetV2_unet(7, 4)