search

MiniBullLab / easy_ai

3 stars 1 forks source link

one class 精度问题 #132

Closed foww-0001 closed 3 years ago

foww-0001 commented 3 years ago

问题是： Memory bank的维度是图片张数feature_map高度feature_map宽度， 正常为163856，easy_ai的结果是150528，大概定位到是

  for index, batch_data in enumerate(self.dataloader):
      print("index: {}".format(index))
      prediction, _ = self.inference.infer(batch_data)
      self.process_func.add_embedding(prediction)
  self.process_func.save_embedding()

图片总张数为209，feature_map高度=feature_map宽度=28。

lpj0822 commented 3 years ago

ok 为0