Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit a3272aad authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

[FasterRCNN] fix box clipping (fix #587)

parent efa492d6
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
examples/FasterRCNN/eval.py
View file @ a3272aad
...@@ -15,7 +15,7 @@ from pycocotools.cocoeval import COCOeval ...@@ -15,7 +15,7 @@ from pycocotools.cocoeval import COCOeval
import pycocotools.mask as cocomask import pycocotools.mask as cocomask
from coco import COCOMeta from coco import COCOMeta
from common import CustomResize from common import CustomResize, clip_boxes
import config import config
DetectionResult = namedtuple( DetectionResult = namedtuple(
...@@ -75,6 +75,8 @@ def detect_one_image(img, model_func): ...@@ -75,6 +75,8 @@ def detect_one_image(img, model_func):
scale = (resized_img.shape[0] * 1.0 / img.shape[0] + resized_img.shape[1] * 1.0 / img.shape[1]) / 2 scale = (resized_img.shape[0] * 1.0 / img.shape[0] + resized_img.shape[1] * 1.0 / img.shape[1]) / 2
boxes, probs, labels, *masks = model_func(resized_img) boxes, probs, labels, *masks = model_func(resized_img)
boxes = boxes / scale boxes = boxes / scale
# boxes are already clipped inside the graph, but after the floating point scaling, this may not be true any more.
boxes = clip_boxes(boxes, orig_shape)
if masks: if masks:
# has mask # has mask
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment