Skip to content

Commit

Permalink
Update mpii_data_upload.py
Browse files Browse the repository at this point in the history
  • Loading branch information
kzuri authored Nov 22, 2020
1 parent 8f3c1c5 commit 0c601bc
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions examples/MPII Human Pose Dataset/mpii_data_upload.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,11 @@ def __call__(self, input):
ds["people_index"] = np.empty(n, object)
ds["numOtherPeople"] = np.empty(n, object)

ds["image"][0] = np.array(Image.open(os.path.join(self._args.dataset_path,"images",input["img_paths"])))
ds["image"][0] = np.array(
Image.open(
os.path.join(self._args.dataset_path,"images",input["img_paths"])
)
)
ds["dataset"][0] = input["dataset"]
ds["isValidation"][0] = input["isValidation"]
ds["img_paths"][0] = input["img_paths"]
Expand Down Expand Up @@ -110,7 +114,7 @@ def load_dataset(args):
in the list(annotations). Finally it returns the complete dataset with all examples.
"""

with open(os.path.join(args.dataset_path,"mpii_annotations.json"), "r") as f:
with open(os.path.join(args.dataset_path, "mpii_annotations.json"), "r") as f:

instances = json.load(f)

Expand Down

0 comments on commit 0c601bc

Please sign in to comment.