The format of the training photos is different

Ysnylmz · May 19, 2019, 10:37am

Hi everyone , all of my training pictures are not the same format.some jpg some jpeg.What do I have to do in this situation.

rohit_gr · May 19, 2019, 3:44pm

Try passing a list of extensions in your dataset.

github.com

fastai/fastai/blob/2830f758a8e3cea7e10b1720a73116c23c295005/fastai/vision/data.py#L275


    "Open image in `fn`, subclass and overwrite for custom behavior."
    return open_image(fn, convert_mode=self.convert_mode, after_open=self.after_open)


def get(self, i):
    fn = super().get(i)
    res = self.open(fn)
    self.sizes[i] = res.size
    return res


@classmethod
def from_folder(cls, path:PathOrStr='.', extensions:Collection[str]=None, **kwargs)->ItemList:
    "Get the list of files in `path` that have an image suffix. `recurse` determines if we search subfolders."
    extensions = ifnone(extensions, image_extensions)
    return super().from_folder(path=path, extensions=extensions, **kwargs)


@classmethod
def from_df(cls, df:DataFrame, path:PathOrStr, cols:IntsOrStrs=0, folder:PathOrStr=None, suffix:str='', **kwargs)->'ItemList':
    "Get the filenames in `cols` of `df` with `folder` in front of them, `suffix` at the end."
    suffix = suffix or ''
    res = super().from_df(df, path=path, cols=cols, **kwargs)
    pref = f'{res.path}{os.path.sep}'