fix

2020-06-24 13:59:02 +08:00 · 2020-06-24 13:59:02 +08:00 · 6891da7612
parent d01581cf80
commit 6891da7612
3 changed files with 3 additions and 0 deletions
--- a/openselfsup/datasets/deepcluster.py
+++ b/openselfsup/datasets/deepcluster.py
@ -1,3 +1,4 @@
+from PIL import Image
 from .registry import DATASETS
 from .base import BaseDataset

--- a/openselfsup/datasets/npid.py
+++ b/openselfsup/datasets/npid.py
@ -1,3 +1,4 @@
+from PIL import Image
 from .registry import DATASETS
 from .base import BaseDataset

--- a/openselfsup/datasets/rotation_pred.py
+++ b/openselfsup/datasets/rotation_pred.py
@ -1,4 +1,5 @@
 import torch
+from PIL import Image

 from .registry import DATASETS
 from .base import BaseDataset