| import torch | |
| from torch.utils.data import Dataset | |
| import os.path as osp | |
| import json | |
| import cv2 | |
| from skimage import io | |
| from PIL import Image | |
| import numpy as np | |
| import random | |
| from torch.utils.data.dataloader import default_collate | |
| from torch.utils.data.dataloader import DataLoader | |
| import matplotlib.pyplot as plt | |
| from torchvision.transforms import functional as F | |
| import copy | |
| # class WireframeDataset(Dataset): | |
| # def __init__(self, ) |