Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Commit cc810aa

Browse files
committed
fix generate testing image list from args.imgs
1 parent 4a80234 commit cc810aa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

‎test.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -185,8 +185,8 @@ def main(cfg, gpu):
185185
os.path.exists(cfg.MODEL.weights_decoder), "checkpoint does not exitst!"
186186

187187
# 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)
190190
else:
191191
imgs = [args.imgs]
192192
assert len(imgs), "imgs should be a path to image (.jpg) or directory."

0 commit comments

Comments
(0)

AltStyle によって変換されたページ (->オリジナル) /