sangyun884 / HR-VITON

Official PyTorch implementation for the paper High-Resolution Virtual Try-On with Misalignment and Occlusion-Handled Conditions (ECCV 2022).
825 stars 171 forks source link

Generate parts of the body #51

Open armangrigoryan96 opened 1 year ago

armangrigoryan96 commented 1 year ago

Dear Authors, How does the network understand that it has to generate hands/neck/hair (when tryin on t-shirt on a long sleeve)? How to reproduce the same result with generating legs (like when trying on skirt on a jeans)?

Should I change something here?

        labels = {
            0:  ['background',  [0]],
            1:  ['paste',       [2, 4, 7, 8, 9, 10, 11]],
            2:  ['upper',       [3]],
            3:  ['hair',        [1]],
            4:  ['left_arm',    [5]],
            5:  ['right_arm',   [6]],
            6:  ['noise',       [12]]
        }