当前位置：网站首页>[Syntax sugar] About the mapping of category strings to category numeric ids

[Syntax sugar] About the mapping of category strings to category numeric ids

2022-08-10 02:30:00 【Master Luwen】

In https://github.com/ming71/DAL, dota_dataset.py:

self.classes = ('__background__', 'plane', 'ship', 'storage-tank', 'baseball-diamond','tennis-court', 'basketball-court', 'ground-track-field', 'harbor','bridge', 'large-vehicle', 'small-vehicle', 'helicopter', 'roundabout','soccer-ball-field' , 'swimming-pool'span>)self.num_classes = len(self.classes)self.class_to_ind = dict(zip(self.classes, range(self.num_classes)))span>