formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--filename',
dest='filename',
help='Image to process',
default='test4.jpg',
type=str)
args = parser.parse_args()
filename = args.filename
img = cv2.imread(filename)
print '####imsize%%', img.shape
t1 = time.time()
boxes, boxes_c = detector.detect_pnet(img)
boxes, boxes_c = detector.detect_rnet(img, boxes_c)
boxes, boxes_c = detector.detect_onet(img, boxes_c)
print('time: %.3fs' % (time.time() - t1))
print('num of boxes:', boxes_c.shape[0])
if boxes_c is not None:
draw = img.copy()
font = cv2.FONT_HERSHEY_SIMPLEX
for b in boxes_c:
cv2.rectangle(draw, (int(b[0]), int(b[1])), (int(b[2]), int(b[3])),
(0, 255, 255), 1)
cv2.putText(draw, '%.3f' % b[4], (int(b[0]), int(b[1])), font, 0.4,
(255, 255, 255), 1)
while True:
cv2.imshow("detection result", draw)
#f = filename.split('.')
