| 1234567891011121314151617181920212223242526 |
- import fastai
- from fastai import *
- from fastai.core import *
- from fastai.vision import *
- def get_colorize_data(sz:int, bs:int, crappy_path:Path, good_path:Path, random_seed:int=None,
- keep_pct:float=1.0, num_workers:int=8)->ImageDataBunch:
- src = (ImageImageList.from_folder(crappy_path)
- .use_partial_data(sample_pct=keep_pct, seed=random_seed)
- .random_split_by_pct(0.1, seed=random_seed))
- data = (src.label_from_func(lambda x: good_path/x.relative_to(crappy_path))
- .transform(get_transforms(max_zoom=1.2, max_lighting=0.5, max_warp=0.25), size=sz, tfm_y=True)
- .databunch(bs=bs, num_workers=num_workers, no_check=True)
- .normalize(imagenet_stats, do_y=True))
- data.c = 3
- return data
- def get_dummy_databunch()->ImageDataBunch:
- path = Path('./dummy/')
- return get_colorize_data(sz=1, bs=1, crappy_path=path, good_path=path, keep_pct=0.001)
|
