Moved synset import

hanrusman · Aug 6, 2019 · 423ffd7 · 423ffd7
1 parent 5cd8051
commit 423ffd7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lucid/modelzoo/vision_base.py b/lucid/modelzoo/vision_base.py
@@ -28,9 +28,6 @@
 from lucid.misc.io import load, save
 import lucid.misc.io.showing as showing
 
-# ImageNet classes correspond to WordNet Synsets.
-from lucid.modelzoo.wordnet import synset_from_id
-
 
 IMAGENET_MEAN = np.array([123.68, 116.779, 103.939])
 IMAGENET_MEAN_BGR = np.flip(IMAGENET_MEAN, 0)
@@ -145,6 +142,8 @@ def synset_ids(self):
 
   @property
   def synsets(self):
+    from lucid.modelzoo.wordnet import synset_from_id
+
     if not self._synsets:
       self._synsets = [synset_from_id(s_id) for s_id in self.synset_ids]
     return self._synsets