You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.3 KiB
53 lines
1.3 KiB
5 years ago
|
#!/usr/bin/env python3.7
|
||
|
|
# pylint: skip-file
|
||
|
|
|
||
|
|
import functools
|
||
|
|
import numpy as np
|
||
|
|
from paddle.fluid.core import AnalysisConfig
|
||
|
|
from paddle.fluid.core import AnalysisPredictor
|
||
|
|
from paddle.fluid.core import create_paddle_predictor
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
config = set_config()
|
||
|
|
predictor = create_paddle_predictor(config)
|
||
|
|
|
||
|
|
data, result = parse_data()
|
||
|
|
|
||
|
|
input_names = predictor.get_input_names()
|
||
|
|
input_tensor = predictor.get_input_tensor(input_names[0])
|
||
|
|
shape = (1, 3, 300, 300)
|
||
|
|
input_data = data[:-4].astype(np.float32).reshape(shape)
|
||
|
|
input_tensor.copy_from_cpu(input_data)
|
||
|
|
|
||
|
|
predictor.zero_copy_run()
|
||
|
|
|
||
|
|
output_names = predictor.get_output_names()
|
||
|
|
output_tensor = predictor.get_output_tensor(output_names[0])
|
||
|
|
output_data = output_tensor.copy_to_cpu()
|
||
|
|
|
||
|
|
|
||
|
|
def set_config():
|
||
|
|
config = AnalysisConfig("")
|
||
|
|
config.set_model("model/__model__", "model/__params__")
|
||
|
|
config.switch_use_feed_fetch_ops(False)
|
||
|
|
config.switch_specify_input_names(True)
|
||
|
|
config.enable_profile()
|
||
|
|
|
||
|
|
return config
|
||
|
|
|
||
|
|
|
||
|
|
def parse_data():
|
||
|
|
""" parse input and output data """
|
||
|
|
with open('data/data.txt', 'r') as fr:
|
||
|
|
data = np.array([float(_) for _ in fr.read().split()])
|
||
|
|
|
||
|
|
with open('data/result.txt', 'r') as fr:
|
||
|
|
result = np.array([float(_) for _ in fr.read().split()])
|
||
|
|
|
||
|
|
return (data, result)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|