Dataset creation issue

[aesanchezgh]

Hi,

When using make_dataset.py on the CelebA dataset, most of my images are failing on the face detection part? I am passing the 512x512 directory. Any idea what might be wrong? The part failing is here:

    def get(self, img, crop_size, max_num=0):
        bboxes, kpss = self.det_model.detect(img,
                                             threshold=self.det_thresh,
                                             max_num=max_num,
                                             metric='default')
        if bboxes.shape[0] == 0:
            return None, None

bboxes is empty and returns None, None for most images.

Thanks!

[kyugorithm]

Hi,

If your detector output None, the there is one choice the resize your detection size from 640 to smaller one.
It can improve your detection results.

[aesanchezgh]

Ok thanks!