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