fix some imports

examples/svhn-digit-convnet.py

@@ -8,15 +8,9 @@ import argparse
 import numpy as np
 import os
 
-from tensorpack.train import TrainConfig, QueueInputTrainer
-from tensorpack.models import *
-from tensorpack.callbacks import *
-from tensorpack.utils import *
-from tensorpack.tfutils import *
+from tensorpack import *
 from tensorpack.tfutils.symbolic_functions import *
 from tensorpack.tfutils.summary import *
-from tensorpack.dataflow import *
-from tensorpack.dataflow import imgaug
 
 """
 SVHN convnet.

opt-requirements.txt

+pillow
 scipy
 nltk
 h5py
 pyzmq
-subprocess32
 tornado

requirements.txt

 termcolor
-pillow
 tqdm
-msgpack
+msgpack-python
 msgpack-numpy
+subprocess32

tensorpack/dataflow/dataset/ilsvrc.py

@@ -17,7 +17,7 @@ CAFFE_ILSVRC12_URL = "http://dl.caffe.berkeleyvision.org/caffe_ilsvrc12.tar.gz"
 CAFFE_PROTO_URL = "https://github.com/BVLC/caffe/raw/master/src/caffe/proto/caffe.proto"
 
 """
-Preprocessing like this:
+Preprocess training set like this:
     cd train
     for i in *.tar; do dir=${i%.tar}; echo $dir; mkdir -p $dir; tar xf $i -C $dir; done
 """

tensorpack/dataflow/dftools.py

@@ -6,6 +6,7 @@ import sys, os
 import cv2
 import multiprocessing
 
+from ..utils.concurrency import DIE
 from ..utils.fs import mkdir_p
 
 __all__ = ['dump_dataset_images', 'dataflow_to_process_queue']

tensorpack/predict/dataset.py

@@ -6,6 +6,8 @@
 from six.moves import range
 from tqdm import tqdm
 from abc import ABCMeta, abstractmethod
+import multiprocessing
+import os
 
 from ..dataflow import DataFlow, BatchData
 from ..dataflow.dftools import dataflow_to_process_queue

tensorpack/utils/utils.py

@@ -91,7 +91,8 @@ def get_dataset_dir(name):
     if d:
         assert os.path.isdir(d)
     else:
-        d = os.path.join(os.path.dirname(__file__), '..', 'dataflow', 'dataset')
+        d = os.path.abspath(os.path.join(
+                    os.path.dirname(__file__), '..', 'dataflow', 'dataset'))
         logger.info("TENSORPACK_DATASET not set, using {} for dataset.".format(d))
     return os.path.join(d, name)