Skip to content

Commit

Permalink
Remove unnecessary function call from within a loop
Browse files Browse the repository at this point in the history
  • Loading branch information
Pavlos Melissinos authored and waleedka committed Dec 10, 2017
1 parent 1bf40f2 commit 1d92c3f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model.py
Original file line number Diff line number Diff line change
Expand Up @@ -753,8 +753,8 @@ def __init__(self, config=None, **kwargs):
def call(self, inputs):
def wrapper(rois, mrcnn_class, mrcnn_bbox, image_meta):
detections_batch = []
_, _, window, _ = parse_image_meta(image_meta)
for b in range(self.config.BATCH_SIZE):
_, _, window, _ = parse_image_meta(image_meta)
detections = refine_detections(
rois[b], mrcnn_class[b], mrcnn_bbox[b], window[b], self.config)
# Pad with zeros if detections < DETECTION_MAX_INSTANCES
Expand Down

0 comments on commit 1d92c3f

Please sign in to comment.