localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

In Compose, replace lambda function with named function (#151)

Browse Source 
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:
Gianluca Gippetto
2021-09-24 03:42:20 +02:00
committed by GitHub
parent 3b473b0e68
commit c13005fd42
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
clip/clip.py
View File

@ -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)),
]) ])