def verify_model(args, model_pb, test_img_file): check_blobs = ["result_boxes", "result_classids"] # result print("Loading test file {}...".format(test_img_file)) test_img = cv2.imread(test_img_file) assert test_img is not None def _run_cfg_func(im, blobs): return run_model_cfg(args, im, check_blobs) def _run_pb_func(im, blobs): return run_model_pb(args, model_pb[0], model_pb[1], im, check_blobs) print("Checking models...") assert mutils.compare_model(_run_cfg_func, _run_pb_func, test_img, check_blobs)
def verify_model(args, model_pb, test_img_file): check_blobs = [ "result_boxes", "result_classids", # result ] print('Loading test file {}...'.format(test_img_file)) test_img = cv2.imread(test_img_file) assert test_img is not None def _run_cfg_func(im, blobs): return run_model_cfg(args, im, check_blobs) def _run_pb_func(im, blobs): return run_model_pb(args, model_pb[0], model_pb[1], im, check_blobs) print('Checking models...') assert mutils.compare_model( _run_cfg_func, _run_pb_func, test_img, check_blobs)
def verify_model(args, model_pb, test_img_file): check_blobs = [ "result_boxes", "result_classids", # result ] workspace.ResetWorkspace() device_opts = core.DeviceOption(caffe2_pb2.CUDA, 1) print('Loading test file {}...'.format(test_img_file)) test_img = cv2.imread(test_img_file) assert test_img is not None def _run_cfg_func(im, blobs): return run_model_cfg(args, im, check_blobs) def _run_pb_func(im, blobs): return run_model_pb(args, model_pb[0], model_pb[1], im, check_blobs) print('Checking models...') assert mutils.compare_model(_run_cfg_func, _run_pb_func, test_img, check_blobs)