Closed lizhen2017 closed 5 years ago
def im_detect(rois, scores, bbox_deltas, im_info, bbox_stds, nms_thresh, conf_thresh): """rois (nroi, 4), scores (nrois, nclasses), bbox_deltas (nrois, 4 * nclasses), im_info (3)""" import time t1 = time.time() rois = rois.asnumpy() print(rois) It takes about 90ms to run rois.asnumpy() Thanks for your help!
def im_detect(rois, scores, bbox_deltas, im_info, bbox_stds, nms_thresh, conf_thresh): """rois (nroi, 4), scores (nrois, nclasses), bbox_deltas (nrois, 4 * nclasses), im_info (3)""" import time t1 = time.time() rois = rois.asnumpy() print(rois) It takes about 90ms to run rois.asnumpy() Thanks for your help!