Fix in apply_net.py dump directory. (facebookresearch#227)

shapovalov · facebook-github-bot · commit 046f035d7e22 · 2019-11-04T11:31:26.000-08:00
Summary: It should use full names after listdir. Thanks for your contribution! If you're sending a large PR (e.g., >50 lines), please open an issue first about the feature / bug, and indicate how you want to contribute. Before submitting a PR, please run `dev/linter.sh` to lint the code. Pull Request resolved: facebookresearch#227 Differential Revision: D18301671 Pulled By: ppwwyyxx fbshipit-source-id: 845778bbc072617652bd20f0ac90a1bc08f34a20
diff --git a/projects/DensePose/apply_net.py b/projects/DensePose/apply_net.py
@@ -101,7 +101,7 @@ def setup_config(
     def _get_input_file_list(cls: type, input_spec: str):
         if os.path.isdir(input_spec):
             file_list = [
-                fname
+                os.path.join(input_spec, fname)
                 for fname in os.listdir(input_spec)
                 if os.path.isfile(os.path.join(input_spec, fname))
             ]

Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,7 @@ def setup_config(`
`101`	`101`	`def _get_input_file_list(cls: type, input_spec: str):`
`102`	`102`	`if os.path.isdir(input_spec):`
`103`	`103`	`file_list = [`
`104`		`- fname`
	`104`	`+ os.path.join(input_spec, fname)`
`105`	`105`	`for fname in os.listdir(input_spec)`
`106`	`106`	`if os.path.isfile(os.path.join(input_spec, fname))`
`107`	`107`	`]`