fix potential bugs

Tengke-Xiong · Nov 16, 2015 · d76c23c · d76c23c
1 parent de7b44b
commit d76c23c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/functions/fast_rcnn/fast_rcnn_im_detect.m b/functions/fast_rcnn/fast_rcnn_im_detect.m
@@ -100,7 +100,7 @@
 
         areas = widths .* heights;
         scaled_areas = bsxfun(@times, areas(:), scales(:)'.^2);
-        levels = max(abs(scaled_areas - 224.^2), 2); 
+        [~, levels] = min(abs(scaled_areas - 224.^2), [], 2); 
     else
         levels = ones(size(im_rois, 1), 1);
     end

diff --git a/functions/fast_rcnn/fast_rcnn_prepare_image_roidb.m b/functions/fast_rcnn/fast_rcnn_prepare_image_roidb.m
@@ -118,7 +118,7 @@
         % Get IoU overlap between each ex ROI and gt ROI
         ex_gt_overlaps = boxoverlap(rois(ex_inds, :), rois(gt_inds, :));
 
-        assert(all(abs(max(ex_gt_overlaps, [], 2) - max_overlaps(ex_inds)) < 1^-6));
+        assert(all(abs(max(ex_gt_overlaps, [], 2) - max_overlaps(ex_inds)) < 10^-4));
 
         % Find which gt ROI each ex ROI has max overlap with:
         % this will be the ex ROI's gt target