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

fix mprnet app (#558)

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