fix bug in InferenceRunner

examples/FasterRCNN/config.py

@@ -166,6 +166,7 @@ _C.MRCNN.HEAD_DIM = 256
 _C.TEST.FRCNN_NMS_THRESH = 0.5
 
 # Smaller threshold value gives significantly better mAP. But we use 0.05 for consistency with Detectron.
+# mAP with 1e-4 threshold can be found at https://github.com/tensorpack/tensorpack/commit/26321ae58120af2568bdbf2269f32aa708d425a8#diff-61085c48abee915b584027e1085e1043  # noqa
 _C.TEST.RESULT_SCORE_THRESH = 0.05
 _C.TEST.RESULT_SCORE_THRESH_VIS = 0.3   # only visualize confident results
 _C.TEST.RESULTS_PER_IM = 100

tensorpack/callbacks/inference_runner.py

@@ -214,7 +214,7 @@ class DataParallelInferenceRunner(InferenceRunnerBase):
         input_callbacks = self._input_source.setup(tower_func.inputs_desc)
         with tf.variable_scope(tf.get_variable_scope(), reuse=True):
             for idx, dev in enumerate(self._devices):
-                vs_name = self._vs_name_for_predictor(idx)
+                vs_name = self.trainer._vs_name_for_predictor(idx)
                 with tf.device(dev), PredictTowerContext(
                         self._tower_names[idx], vs_name=vs_name):
                     logger.info("[InferenceRunner] Building tower '{}' on device {} {}...".format(

tensorpack/dataflow/parallel.py

@@ -231,11 +231,12 @@ class PrefetchDataZMQ(_MultiProcessZMQDataFlow):
            Please note that forking a TensorFlow GPU session may be unsafe.
            If you're managing this dataflow on your own,
            it's better to fork before creating the session.
-        4. After the fork has happened, this dataflow becomes not fork-safe.
+        4. (Fork-safety) After the fork has happened, this dataflow becomes not fork-safe.
            i.e., if you fork an already reset instance of this dataflow,
-           it won't be usable in the forked process.
-        5. Do not nest two `PrefetchDataZMQ`.
-        6. By default, a UNIX named pipe will be created in the current directory.
+           it won't be usable in the forked process. Therefore, do not nest two `PrefetchDataZMQ`.
+        5. (Thread-safety) ZMQ is not thread safe. Therefore, do not call :meth:`get_data` of the same dataflow in
+           more than 1 threads.
+        6. (For Mac only) A UNIX named pipe will be created in the current directory.
            However, certain non-local filesystem such as NFS/GlusterFS/AFS doesn't always support pipes.
            You can change the directory by ``export TENSORPACK_PIPEDIR=/other/dir``.
            In particular, you can use somewhere under '/tmp' which is usually local.
@@ -428,6 +429,8 @@ class PlasmaGetData(ProxyDataFlow):
     """
     Take plasma object id from a DataFlow, and retrieve it from plasma shared
     memory object store.
+
+    Experimental.
     """
     def __init__(self, ds, socket="/tmp/plasma"):
         self._socket = socket