52b334a63ea83ef5a6a4033caf3ae76cb6fed27f,torchreid/data/datasets/dataset.py,ImageDataset,show_summary,#ImageDataset#,290

Before Change

        return item

    def show_summary(self):
        num_train_pids, num_train_cams = self.parse_data(self.train)
        num_query_pids, num_query_cams = self.parse_data(self.query)
        num_gallery_pids, num_gallery_cams = self.parse_data(self.gallery)

After Change

    def show_summary(self):
        num_train_pids = self.get_num_pids(self.train)
        num_train_cams = self.get_num_cams(self.train)

        num_query_pids = self.get_num_pids(self.query)
        num_query_cams = self.get_num_cams(self.query)

        num_gallery_pids = self.get_num_pids(self.gallery)
        num_gallery_cams = self.get_num_cams(self.gallery)

        print("=> Loaded {}".format(self.__class__.__name__))
        print("  ----------------------------------------")
        print("  subset   | // ids | // images | // cameras")

Instances