未验证 提交 3e17a8c2 编写于 作者: L LielinJiang 提交者: GitHub

fix mprnet app (#558)

上级 d77ae353
......@@ -59,9 +59,9 @@ if __name__ == "__main__":
if args.cpu:
paddle.set_device('cpu')
predictor = MPRPredictor(images_path=args.images_path,
predictor = MPRPredictor(
output_path=args.output_path,
weight_path=args.weight_path,
seed=args.seed,
task=args.task)
predictor.run()
predictor.run(images_path=args.images_path)
......@@ -89,6 +89,7 @@ class MPRPredictor(BasePredictor):
def get_images(self, images_path):
if os.path.isdir(images_path):
return natsorted(
glob(os.path.join(images_path, '*.jpeg')) +
glob(os.path.join(images_path, '*.jpg')) +
glob(os.path.join(images_path, '*.JPG')) +
glob(os.path.join(images_path, '*.png')) +
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册