mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
30 lines
808 B
Python
30 lines
808 B
Python
import os
|
|
import sys
|
|
__dir__ = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
|
|
|
|
import cv2
|
|
|
|
from engine import build_engine
|
|
from utils import config
|
|
from utils.get_image_list import get_image_list
|
|
|
|
|
|
def main():
|
|
args = config.parse_args()
|
|
config_dict = config.get_config(
|
|
args.config, overrides=args.override, show=False)
|
|
config_dict.profiler_options = args.profiler_options
|
|
engine = build_engine(config_dict)
|
|
|
|
image_list = get_image_list(config_dict["Global"]["infer_imgs"])
|
|
for idx, image_file in enumerate(image_list):
|
|
img = cv2.imread(image_file)[:, :, ::-1]
|
|
input_data = {"input_image": img}
|
|
output = engine.process(input_data)
|
|
print(output)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|