From a6a302c98ec5fee620a90e669bd9974935256355 Mon Sep 17 00:00:00 2001 From: michaelzhaobin <31604328+michaelzhaobin@users.noreply.github.com> Date: Sun, 4 Mar 2018 21:20:58 +0800 Subject: [PATCH] Update minibatch.py --- lib/roi_data_layer/minibatch.py | 1 + 1 file changed, 1 insertion(+) 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)