paddlex_image_classification_pipeline_quickstart_inference.py

1from paddlex import create_pipeline
2
3# Initialize the pipeline (using Image Classification as an example)
4pipeline = create_pipeline(pipeline_name="cls")
5
6# Perform inference on an image (URL or local path)
7output = pipeline.predict(input_data="https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_classification_001.jpg")
8
9# Iterate through the generator results and print them
10for res in output:
11    res.print()
12    res.save_to_img("./output/")
13    res.save_to_json("./output/res.json")