Error testing the example provided about neuralhaircut

[carlosedubarreto]

I think I've foud the problem, the values ar duplicating

Now I'll try to find why ☺️

[carlosedubarreto]

Found a solution changin a bit the information where it loads the images

    # self.images_lis = sorted(glob_imgs(os.path.join(self.data_dir, 'image')))
    self.images_lis = sorted(glob(os.path.join(self.data_dir, 'image','*.png')))

    # self.masks_lis = sorted(glob_imgs(os.path.join(self.data_dir, 'mask')))
    self.masks_lis = sorted(glob(os.path.join(self.data_dir, 'mask','*.png')))

load hair mask

    # self.hair_masks_lis = sorted(glob_imgs(os.path.join(self.data_dir, 'hair_mask')))
    self.hair_masks_lis = sorted(glob(os.path.join(self.data_dir, 'hair_mask','*.png')))

load orientations

    # self.orientations_lis = sorted(glob_imgs(os.path.join(self.data_dir, 'orientation_maps')))
    self.orientations_lis = sorted(glob(os.path.join(self.data_dir, 'orientation_maps','*.png')))

load variance

    # self.variance_lis = sorted(glob_imgs(os.path.join(self.data_dir, 'confidence_maps')))
    self.variance_lis = sorted(glob(os.path.join(self.data_dir, 'confidence_maps','*.png')))

[egorzakharov]

Hello Carlos,

Thanks for investigating this issue! We'll take a look at it.

[carlosedubarreto]

Oh I missed one thing, on the last part of the code I wrote it wrong and Vanessa corrected me.

Instead of
self.variance_lis = sorted(glob(os.path.join(self.data_dir, 'confidence_maps','*.png')))

it should be
self.variance_lis = sorted(glob(os.path.join(self.data_dir, 'confidence_maps','*.npy')))