30 lines
856 B
Python
Executable File
30 lines
856 B
Python
Executable File
#!/usr/bin/env python3
|
|
from PIL import Image
|
|
import os
|
|
import argparse
|
|
import sys
|
|
from roi_lib import *
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description='ROI prediction')
|
|
parser.add_argument('--model', type=str, required=True, help='model path')
|
|
parser.add_argument('--image', type=str, required=True, help='image file')
|
|
return parser.parse_args()
|
|
|
|
# 主函数
|
|
def main():
|
|
args = parse_args()
|
|
model = load_model(args.model)
|
|
image_path = args.image
|
|
image_tensor = preprocess_image(image_path)
|
|
predicted_class, probabilities = predict(model, image_tensor)
|
|
print(f'{image_path} predicted={predicted_class} prob={probabilities}')
|
|
if predicted_class == 1:
|
|
print("verify ok")
|
|
return 0
|
|
else:
|
|
print("verify ng")
|
|
return 1
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main()) |