bianbu-repo-mirror/k1x-ai-support
1
0
Fork 0
mirror of https://gitee.com/bianbu-linux/ai-support synced 2025-04-25 06:37:17 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update for v1.0

Browse source
This commit is contained in:
James Deng 2024-05-30 23:19:44 +08:00
parent 46cf2741d2
commit adf80d91df
76 changed files with 1487 additions and 716 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/python/demo_image_classification.py Normal file
View file

@ -0,0 +1,23 @@
import argparse
def get_argsparser():
"""Parse commandline."""
parser = argparse.ArgumentParser(description="Bianbu AI Python Demo for Image Classification.")
parser.add_argument("--image", "-i", type=str, required=True, help="input test image path")
parser.add_argument("--model", "-m", type=str, required=True, help="input test model(*.onnx) path")
parser.add_argument("--label", "-l", type=str, required=True, help="input test label path")
parser.add_argument("--intra", type=int, default=2, help="intra thread number for backend(e.g. onnxruntime)")
parser.add_argument("--inter", type=int, default=2, help="inter thread number for backend(e.g. onnxruntime)")
return parser.parse_args()
if __name__ == "__main__":
args = get_argsparser()
import cv2
from bianbuai import ImageClassificationOption, ImageClassificationTask
option = ImageClassificationOption(args.model, args.label, args.intra, args.inter)
task = ImageClassificationTask(option)
output = task.Classify(cv2.imread(args.image))
print("Label:", output.label_text)
print("Score:", output.score)