def extract_proposal(im_file): im = cv2.imread(im_file)[::-1, :, ::-1] #[:, :, ::-1] if sess_tuple is None: _init() boxes = im_proposal_tensorflow(sess_tuple, im) for box in boxes: im = draw_box(box, im) cv2.imwrite('test.jpg', im) return boxes
def extract_proposal(self, image): boxes = im_proposal_tensorflow(sess_tuple, image) return boxes
def extract_proposal(im_file): im = cv2.imread(im_file) boxes = im_proposal_tensorflow(sess_tuple, im) return boxes