search

simonw / cougar-or-not

An API for identifying cougars v.s. bobcats v.s. other USA cat species
https://cougar-or-not.now.sh/
122 stars 56 forks source link

missing normalization #1

Open francescogianferraripini opened 5 years ago

francescogianferraripini commented 5 years ago

I think you should add normalization cat_data = ImageDataBunch.from_name_re( cat_images_path, catfnames, r"/([^/]+)\d+.jpg$", ds_tfms=get_transforms(), size=299, ).normalize(imagenet_stats)

siAyush commented 5 years ago

cat_images_path = Path('/home/jupyter/.fastai/data/inaturalist-usa-cats/images') cat_fnames = get_image_files(cat_images_path) cat_data = ImageDataBunch.from_name_re( cat_images_path, catfnames, r'/([^/]+)\d+.jpg$', ds_tfms=get_transforms(), size=224) cat_data.normalize(imagenet_stats) Already added