diff --git a/lib/roi_data_layer/minibatch.py b/lib/roi_data_layer/minibatch.py index 674ef209..b78111e3 100644 --- a/lib/roi_data_layer/minibatch.py +++ b/lib/roi_data_layer/minibatch.py @@ -16,6 +16,7 @@ def get_minibatch(roidb, num_classes): """Given a roidb, construct a minibatch sampled from it.""" num_images = len(roidb) + # # Sample random scales to use for each image in this batch random_scale_inds = npr.randint(0, high=len(cfg.TRAIN.SCALES), size=num_images)