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)
|