[FasterRCNN] disable cropsize bug and disable fork

examples/FasterRCNN/data.py

@@ -267,7 +267,7 @@ def get_eval_dataflow():
         assert im is not None, fname
         return im
     ds = MapDataComponent(ds, f, 0)
-    ds = PrefetchDataZMQ(ds, 1)
+    # ds = PrefetchDataZMQ(ds, 1)
     return ds
 
 

examples/FasterRCNN/model.py

@@ -306,6 +306,7 @@ def crop_and_resize(image, boxes, box_ind, crop_size):
     Returns:
         n,C,size,size
     """
+    assert isinstance(crop_size, int), crop_size
     @under_name_scope()
     def transform_fpcoor_for_tf(boxes, image_shape, crop_shape):
         """

examples/FasterRCNN/train.py

@@ -119,7 +119,7 @@ class Model(ModelDesc):
             with tf.name_scope('fg_sample_patch_viz'):
                 fg_sampled_patches = crop_and_resize(
                     image, fg_sampled_boxes,
-                    tf.zeros_like(fg_inds_wrt_sample, dtype=tf.int32), [300, 300])
+                    tf.zeros_like(fg_inds_wrt_sample, dtype=tf.int32), 300)
                 fg_sampled_patches = tf.transpose(fg_sampled_patches, [0, 2, 3, 1])
                 tf.summary.image('viz', fg_sampled_patches, max_outputs=30)
 
@@ -308,7 +308,7 @@ if __name__ == '__main__':
 
         cfg = TrainConfig(
             model=Model(),
-            dataflow=get_train_dataflow(),
+            data=QueueInput(get_train_dataflow()),
             callbacks=[
                 PeriodicTrigger(ModelSaver(), every_k_epochs=5),
                 # linear warmup