diff --git a/Code/Three_view_to_isometric/Dataloader.py b/Code/Three_view_to_isometric/Dataloader.py index 956bcd2..9aece56 100644 --- a/Code/Three_view_to_isometric/Dataloader.py +++ b/Code/Three_view_to_isometric/Dataloader.py @@ -6,6 +6,7 @@ import os import numpy as np import cv2 +import json class ThreeV2I_BC_data(Dataset): def __init__(self, root_dir): self.dic=sorted(os.listdir(root_dir))