13 lines
365 B
Python
13 lines
365 B
Python
import argparse
|
|
from models import ClassificationModel
|
|
import cv2 as cv
|
|
|
|
def main(args):
|
|
cls_model = ClassificationModel()
|
|
print(cls_model.predict(cv.imread(args.image_path)))
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--image_path", type=str, required=True)
|
|
args = parser.parse_args()
|
|
main(args) |