Skip to content

GitLab

"...composable_kernel_rocm.git" did not exist on "3624dc2a7be1159ca61c3ef22e5804d8b752a287"
Commit 4b3ecfb3 authored by Rayyyyy's avatar Rayyyyy
Browse files

Mpdify PATHS in coco.py

parent ae4d41f9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
datasets/coco.py
View file @ 4b3ecfb3
...@@ -149,8 +149,8 @@ def build(image_set, args): ...@@ -149,8 +149,8 @@ def build(image_set, args):
assert root.exists(), f'provided COCO path {root} does not exist' assert root.exists(), f'provided COCO path {root} does not exist'
mode = 'instances' mode = 'instances'
PATHS = { PATHS = {
"train": (root / "images/train2017", root / "annotations" / f'{mode}_train2017.json'), "train": (root / "train2017", root / "annotations" / f'{mode}_train2017.json'),
"val": (root / "images/val2017", root / "annotations" / f'{mode}_val2017.json'), "val": (root / "val2017", root / "annotations" / f'{mode}_val2017.json'),
} }
img_folder, ann_file = PATHS[image_set] img_folder, ann_file = PATHS[image_set]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment