Python pre示例

示例#1

def apiRun(params):
	model = models.SSDMOBILENET
	params = common.pre(model, params)

	modelFileName='ssdmobilenet_'+params[const.PRECISION] + '.xml'
    
	if params == None:
		return None, None
	if params[const.HARDWARE] == const.GPU:
		modelFileName='ssdmobilenet_'+params[const.PRECISION]+'_b'+str(params[const.BATCH])+'.xml'
		if os.path.exists(modelFileName):
			print("Cannot find model file {}. Please ensure an IR was generated for the requested batch size and precision during compilation.".format(modelFileName))
			sys.exit(1)
	else:
		if os.path.exists(modelFileName):
			print("Cannot find model file {}. Please ensure an IR was generated for the requested precision during compilation.".format(modelFileName))
			sys.exit(1)

	if params[const.MODE] in [const.THROUGHPUT, const.LATENCY]:
		net = SSDMOBILENET(
			os.path.join(paths.MODELS, 'openvino', model, modelFileName),
			params[const.HARDWARE], params[const.PRECISION], params[const.MODE], params[const.BATCH], params[const.CONCURRENCY])
	else:
		net = SSDMOBILENET(
			os.path.join(paths.MODELS, 'openvino', model,modelFileName),
			params[const.HARDWARE], params[const.PRECISION], params[const.MODE], params[const.BATCH] )
	return params, common.run(model, net, params)

示例#2

def apiRun(params):
	model = models.SSDMOBILENET
	params = common.pre(model, params)
	if params == None:
		return None, None
	net = SsdMobileNet(
		os.path.join(paths.MODELS, 'tensorflow_lite_quantised', model, 'quantised_PTIQ_edgetpu.tflite'))
	return params, common.run(model, net, params)

示例#3

def apiRun(params):
	model = models.SSDMOBILENET
	params = common.pre(model, params)
	if params == None:
		return None, None
	net = SsdMobileNet(
		os.path.join(paths.MODELS, 'tensorflow', model, 'frozen_graph.pb'),
		params[const.HARDWARE])
	return params, common.run(model, net, params)

示例#4

def apiRun(params):
    model = models.RESNET50
    params = common.pre(model, params)
    if params == None:
        return None, None
    net = ResNet50(
        os.path.join(paths.MODELS, 'tensorflow', model, 'frozen_graph.pb'),
        params[const.HARDWARE])
    return params, common.run(model, net, params)

示例#5

def apiRun(params):
        model = models.MOBILENET
        params = common.pre(model, params)
        if params == None:
                return None, None
        if params[const.MODE] in [const.THROUGHPUT, const.LATENCY]:
                net = MobileNet(os.path.join(paths.MODELS, 'tensorflow_uff', model, 'frozen_graph_mobilenet.uff'),params[const.BATCH],params[const.HARDWARE],params[const.MODE],params[const.PRECISION],params[const.CONCURRENCY])
        else:
                net = MobileNet(os.path.join(paths.MODELS, 'tensorflow_uff', model, 'frozen_graph_mobilenet.uff'),params[const.BATCH],params[const.HARDWARE],params[const.MODE],params[const.PRECISION])
        return params, common.run(model, net, params)

示例#6

def apiRun(params):
	model = models.RESNET50
	params = common.pre(model, params)
	if params == None:
		return None, None
	if params[const.PRECISION] == const.FP32:
		net = ResNet50(os.path.join(paths.MODELS, 'tensorflow_lite', model, 'converted_model.tflite'),params[const.HARDWARE])
		return params, common.run(model, net, params)
	if params[const.PRECISION] == const.INT8:
		net = ResNet50(os.path.join(paths.MODELS, 'tensorflow_lite_quantised', model, 'quantised_PTIQ.tflite'),params[const.HARDWARE])
		return params, common.run(model, net, params)

示例#7

def apiRun(params):
    model = models.RESNET50
    params = common.pre(model, params)
    if params == None:
        return None, None
    if params[const.MODE] in [const.THROUGHPUT, const.LATENCY]:
        net = ResNet50(
            os.path.join(paths.MODELS, 'openvino', model,
                         'resnet-50_' + params[const.PRECISION] + '.xml'),
            params[const.HARDWARE], params[const.PRECISION],
            params[const.MODE], params[const.BATCH], params[const.CONCURRENCY])
    else:
        net = ResNet50(
            os.path.join(paths.MODELS, 'openvino', model,
                         'resnet-50_' + params[const.PRECISION] + '.xml'),
            params[const.HARDWARE], params[const.PRECISION],
            params[const.MODE], params[const.BATCH])
    return params, common.run(model, net, params)

示例#8

def apiRun(params):
    model = models.MOBILENET
    params = common.pre(model, params)
    if params == None:
        return None, None

    if params[const.MODE] in [const.THROUGHPUT, const.LATENCY]:
        net = MobileNet(
            os.path.join(paths.MODELS, 'openvino', model,
                         'mobilenetv1_' + params[const.PRECISION] + '.xml'),
            params[const.HARDWARE], params[const.PRECISION],
            params[const.MODE], params[const.BATCH], params[const.CONCURRENCY])
    else:
        net = MobileNet(
            os.path.join(paths.MODELS, 'openvino', model,
                         'mobilenetv1_' + params[const.PRECISION] + '.xml'),
            params[const.HARDWARE], params[const.PRECISION],
            params[const.MODE], params[const.BATCH])
    return params, common.run(model, net, params)