xthan/VITON

keyerror: Segment

Opened this issue · 1 comments

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

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