Commit cc810aa

committed

fix generate testing image list from args.imgs

1 parent 4a80234 commit cc810aaCopy full SHA for cc810aa

File tree

-2

lines changed

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -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)