17 lines
485 B
Python
17 lines
485 B
Python
import numpy as np
|
|
|
|
from object_detection import CornerNet_Saccade
|
|
from util import image_util
|
|
|
|
|
|
def capture_target_area(image, target="book"):
|
|
detector = CornerNet_Saccade()
|
|
bboxes = detector(image)
|
|
target_images = []
|
|
keep_inds = bboxes[target][:, -1] > 0.5
|
|
for bbox in bboxes[target][keep_inds]:
|
|
bbox = bbox[0:4].astype(np.int32)
|
|
bbox = np.clip(bbox, 0, None)
|
|
target_images.append(image_util.capture(image, bbox))
|
|
return target_images
|