search

bubbliiiing / Siamese-pytorch

这是一个孪生神经网络（Siamese network）的库，可进行图片的相似性比较。
MIT License
539 stars 120 forks source link

dataloader SiameseDataset types 类别缺少最后一类 ？ #38

Closed se202242 closed 11 months ago

se202242 commented 11 months ago

utils.dataloader.SiameseDataset self.types = max(labels) 后续getitem 函数中 c = random.randint(0, self.types - 1)

label = self.types 类是否一直未被选中

更改为 self.types = max(labels) + 1 程序仍跑通

bubbliiiing commented 11 months ago

randint是包括self.types - 1的嘞

se202242 commented 11 months ago

但 labels 中 max 所对应类别不就一直未被取到吗？

bubbliiiing commented 11 months ago

self.types - 1就是最大对应的