fix: fix type of label to int64

ppcls/data/dataloader/icartoon_dataset.py

@@ -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])

ppcls/data/dataloader/imagenet_dataset.py

@@ -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])

ppcls/data/dataloader/logo_dataset.py

@@ -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])
-
-                

ppcls/data/dataloader/multilabel_dataset.py

@@ -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])

ppcls/data/dataloader/vehicle_dataset.py

@@ -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):