search

xthan / VITON

Code and dataset for paper "VITON: An Image-based Virtual Try-on Network"
534 stars 143 forks source link

keyerror: Segment #50

Open lokesh0606 opened 2 years ago

lokesh0606 commented 2 years ago

Getting Keyerror: segment.

segment_raw = sio.loadmat(os.path.join( tf.app.flags.FLAGS.segment_dir, image_id))['segment'] #getting the error on this line segment_raw = process_segment_map(segment_raw, image.shape[0], image.shape[1]) pose_raw = sio.loadmat(os.path.join(FLAGS.pose_dir, image_id))

What would have been the issue and how to resolve it:?

xthan commented 2 years ago

Do you have the segmentation files downloaded in the tf.app.flags.FLAGS.segment_dir? You need to download the pose and segmentations from the Google Drive https://drive.google.com/drive/folders/1bMRMZNbZnX2H5BnYbl63h2vGii3E0e3D?usp=sharing