i need to make a dataset for object localization how it possible?
def show_landmarks(image, landmarks):
“”“Show image with landmarks”""
xmin=landmarks[0]
ymin=landmarks[1]
xmax=landmarks[2]
ymax=landmarks[3]
start=(xmin,ymin)
end=(xmax,ymax)
image=cv2.rectangle(image,start,end,(100,210,69),1)
plt.imshow(image)
class Dataset(torch.utils.data.Dataset):
def __init__(self, csv_file, root_dir, transform=None):
    
    self.csv = pd.read_csv(csv_file)
    self.root   = root_dir
    self.transform = transform

def __len__(self):
    return len(self.csv)

def __getitem__(self,idx):
    
    img_name=self.csv.iloc[idx,0]
    landmarks =self.csv.iloc[idx, 4:].as_matrix()
    
    
    
   
  
    
    
    return img_name,landmarks