- Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathMyData.py
More file actions
Latest commit
38 lines (29 loc) · 920 Bytes
/
Copy pathMyData.py
File metadata and controls
38 lines (29 loc) · 920 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
importos
importrandom
importnumpyasnp
importtorch.utils.dataasdata
importtorchvision.transformsastransforms
fromPILimportImage
classMyData(data.Dataset):
def__init__(self, txt_path, transform=None, target_transform=None):
fh=open(txt_path, 'r')
imgs= []
forlineinfh:
line=line.rstrip()
words=line.split()
imgs.append((words[0], int(words[1])))
self.imgs=imgs
self.transform=transform
self.target_transform=target_transform
def__getitem__(self, index):
fn, label=self.imgs[index]
# print(fn)
# print(label)
img=Image.open(fn).convert('RGB')
ifself.transformisnotNone:
img=self.transform(img)
returnimg, label
def__len__(self):
returnlen(self.imgs)
if__name__=="__main__":
pass