In Compose, replace lambda function with named function (#151)
This prevents the following error on Windows (when using a multi-process DataLoader, for example): AttributeError: Can't pickle local object '_transform.<locals>.<lambda>'
This commit is contained in:
parent
3b473b0e68
commit
c13005fd42
|
@ -68,11 +68,15 @@ def _download(url: str, root: str):
|
||||||
return download_target
|
return download_target
|
||||||
|
|
||||||
|
|
||||||
|
def _convert_image_to_rgb(image):
|
||||||
|
return image.convert("RGB")
|
||||||
|
|
||||||
|
|
||||||
def _transform(n_px):
|
def _transform(n_px):
|
||||||
return Compose([
|
return Compose([
|
||||||
Resize(n_px, interpolation=BICUBIC),
|
Resize(n_px, interpolation=BICUBIC),
|
||||||
CenterCrop(n_px),
|
CenterCrop(n_px),
|
||||||
lambda image: image.convert("RGB"),
|
_convert_image_to_rgb,
|
||||||
ToTensor(),
|
ToTensor(),
|
||||||
Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711)),
|
Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711)),
|
||||||
])
|
])
|
||||||
|
|
Loading…
Reference in New Issue