Skip to content
Snippets Groups Projects
Commit b8599537 authored by liuxingyu's avatar liuxingyu
Browse files

delete assertation for xyz_path

parent acb11494
No related branches found
No related tags found
No related merge requests found
...@@ -190,7 +190,6 @@ class HB_PBR_Dataset: ...@@ -190,7 +190,6 @@ class HB_PBR_Dataset:
self.dataset_root, self.dataset_root,
f"{scene_id:06d}/coor_backprj/{int_im_id:06d}_{anno_i:06d}.pkl", f"{scene_id:06d}/coor_backprj/{int_im_id:06d}_{anno_i:06d}.pkl",
) )
# assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
"bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib "bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib
......
...@@ -177,7 +177,6 @@ class ICBIN_PBR_Dataset: ...@@ -177,7 +177,6 @@ class ICBIN_PBR_Dataset:
self.xyz_root, self.xyz_root,
f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
......
...@@ -186,7 +186,6 @@ class ITODD_PBR_Dataset: ...@@ -186,7 +186,6 @@ class ITODD_PBR_Dataset:
self.xyz_root, self.xyz_root,
f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
# assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
"bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib "bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib
......
...@@ -186,7 +186,6 @@ class LM_PBR_Dataset: ...@@ -186,7 +186,6 @@ class LM_PBR_Dataset:
self.xyz_root, self.xyz_root,
f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
"bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib "bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib
......
...@@ -177,7 +177,6 @@ class TLESS_PBR_Dataset: ...@@ -177,7 +177,6 @@ class TLESS_PBR_Dataset:
self.xyz_root, self.xyz_root,
f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
# assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
......
...@@ -186,7 +186,6 @@ class TUDL_PBR_Dataset: ...@@ -186,7 +186,6 @@ class TUDL_PBR_Dataset:
self.xyz_root, self.xyz_root,
f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
# assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
"bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib "bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib
......
...@@ -247,7 +247,6 @@ class YCBV_Dataset: ...@@ -247,7 +247,6 @@ class YCBV_Dataset:
xyz_root, xyz_root,
f"{scene_id:06d}/{im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
# assert osp.exists(xyz_path), xyz_path
inst["xyz_path"] = xyz_path inst["xyz_path"] = xyz_path
model_info = self.models_info[str(obj_id)] model_info = self.models_info[str(obj_id)]
......
...@@ -187,7 +187,6 @@ class YCBV_PBR_Dataset: ...@@ -187,7 +187,6 @@ class YCBV_PBR_Dataset:
self.xyz_root, self.xyz_root,
f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl", f"{scene_id:06d}/{int_im_id:06d}_{anno_i:06d}-xyz.pkl",
) )
# assert osp.exists(xyz_path), xyz_path
inst = { inst = {
"category_id": cur_label, # 0-based label "category_id": cur_label, # 0-based label
"bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib "bbox": bbox_obj, # TODO: load both bbox_obj and bbox_visib
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment