mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix: fix type of label to int64

pull/1354/head
gaotingquan 2021-10-28 07:09:41 +00:00 committed by Tingquan Gao
parent fbe8880764
commit d3a1a1fc45
5 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/data/dataloader/icartoon_dataset.py
View File

@ -32,5 +32,5 @@ class ICartoonDataset(CommonDataset):
for l in lines:
l = l.strip().split("\t")
self.images.append(os.path.join(self._img_root, l[0]))
self.labels.append(int(l[1]))
self.labels.append(np.int64(l[1]))
assert os.path.exists(self.images[-1])

2
ppcls/data/dataloader/imagenet_dataset.py
View File

@ -34,5 +34,5 @@ class ImageNetDataset(CommonDataset):
for l in lines:
l = l.strip().split(" ")
self.images.append(os.path.join(self._img_root, l[0]))
self.labels.append(int(l[1]))
self.labels.append(np.int64(l[1]))
assert os.path.exists(self.images[-1])

5
ppcls/data/dataloader/logo_dataset.py
View File

@ -28,6 +28,7 @@ import random
from .common_dataset import CommonDataset
class LogoDataset(CommonDataset):
def _load_anno(self):
assert os.path.exists(self._cls_path)
@ -41,7 +42,5 @@ class LogoDataset(CommonDataset):
if l[0] == 'image_id':
continue
self.images.append(os.path.join(self._img_root, l[3]))
self.labels.append(int(l[1])-1)
self.labels.append(np.int64(l[1]) - 1)
assert os.path.exists(self.images[-1])

2
ppcls/data/dataloader/multilabel_dataset.py
View File

@ -37,7 +37,7 @@ class MultiLabelDataset(CommonDataset):
self.images.append(os.path.join(self._img_root, l[0]))
labels = l[1].split(',')
labels = [int(i) for i in labels]
labels = [np.int64(i) for i in labels]
self.labels.append(labels)
assert os.path.exists(self.images[-1])

4
ppcls/data/dataloader/vehicle_dataset.py
View File

@ -112,8 +112,8 @@ class VeriWild(Dataset):
for l in lines:
l = l.strip().split()
self.images.append(os.path.join(self._img_root, l[0]))
self.labels.append(int(l[1]))
self.cameras.append(int(l[2]))
self.labels.append(np.int64(l[1]))
self.cameras.append(np.int64(l[2]))
assert os.path.exists(self.images[-1])
def __getitem__(self, idx):