Commit 69243f0

authored

Merge pull request #228 from yagi-3/fix_test_argument_imgs

Fix argument imgs in test

2 parents 7a68e75 + 4d7cb68 commit 69243f0Copy full SHA for 69243f0

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -141,7 +141,7 @@ def main(cfg, gpu):`
`141`	`141`	`"--imgs",`
`142`	`142`	`required=True,`
`143`	`143`	`type=str,`
`144`		`- help="an image paths, or a directory name"`
	`144`	`+ help="an image path, or a directory name"`
`145`	`145`	`)`
`146`	`146`	`parser.add_argument(`
`147`	`147`	`"--cfg",`
`@@ -185,8 +185,8 @@ def main(cfg, gpu):`
`185`	`185`	`os.path.exists(cfg.MODEL.weights_decoder), "checkpoint does not exitst!"`
`186`	`186`
`187`	`187`	`# generate testing image list`
`188`		`- if os.path.isdir(args.imgs[0]):`
`189`		`- imgs = find_recursive(args.imgs[0])`
	`188`	`+ if os.path.isdir(args.imgs):`
	`189`	`+ imgs = find_recursive(args.imgs)`
`190`	`190`	`else:`
`191`	`191`	`imgs = [args.imgs]`
`192`	`192`	`assert len(imgs), "imgs should be a path to image (.jpg) or directory."`

Comments

(0)