Solved by Jordan_Howell in post #5 
              
              
                So, I think I have the class working correctly… 

class roofDataset(Dataset): 
‘’’ 
roof data class 
'''
def __init__(self, csv_file, transform = None):
    '''
    Args:
        csv_file (string): Path to the csv file with annotations.
        transform (callable, optional): Optional transform to b…