python3 compat

docs/conf.py

@@ -21,7 +21,7 @@ import os
 sys.path.insert(0, os.path.abspath('../'))
 
 import mock
-MOCK_MODULES = ['numpy', 'scipy', 'tensorflow', 'scipy.misc', 'h5py', 'nltk']
+MOCK_MODULES = ['numpy', 'scipy', 'tensorflow', 'scipy.misc', 'h5py', 'nltk', 'cv2']
 for mod_name in MOCK_MODULES:
     sys.modules[mod_name] = mock.Mock()
 

tensorpack/callbacks/inference.py

@@ -5,7 +5,7 @@
 import tensorflow as tf
 from tqdm import tqdm
 from abc import ABCMeta, abstractmethod
-from six.moves import zip
+from six.moves import zip, map
 
 from ..dataflow import DataFlow
 from ..utils import *
@@ -102,7 +102,7 @@ class InferenceRunner(Callback):
         def get_tensor(name):
             _, varname = get_op_var_name(name)
             return self.graph.get_tensor_by_name(varname)
-        self.output_tensors = map(get_tensor, self.output_tensors)
+        self.output_tensors = list(map(get_tensor, self.output_tensors))
 
     def _trigger_epoch(self):
         for vc in self.vcs:

tensorpack/dataflow/common.py

@@ -2,8 +2,9 @@
 # File: common.py
 # Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-import numpy as np
+from __future__ import division
 import copy
+import numpy as np
 from six.moves import range
 from .base import DataFlow, ProxyDataFlow
 from ..utils import *
@@ -30,7 +31,7 @@ class BatchData(ProxyDataFlow):
 
     def size(self):
         ds_size = self.ds.size()
-        div = ds_size / self.batch_size
+        div = ds_size // self.batch_size
         rem = ds_size % self.batch_size
         if rem == 0:
             return div

tensorpack/dataflow/dataset/visualqa.py

@@ -73,7 +73,7 @@ if __name__ == '__main__':
     vqa = VisualQA('/home/wyx/data/VQA/MultipleChoice_mscoco_train2014_questions.json',
             '/home/wyx/data/VQA/mscoco_train2014_annotations.json')
     for k in vqa.get_data():
-        print json.dumps(k)
+        print(json.dumps(k))
         break
 #    vqa.get_common_question_words(100)
     vqa.get_common_answer(100)