当前位置：网站首页>[语法糖] 关于类别字符串到类别数字id的映射

[语法糖] 关于类别字符串到类别数字id的映射

2022-08-10 01:12:00 【氵文大师】

在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')   
self.num_classes = len(self.classes)
self.class_to_ind = dict(zip(self.classes, range(self.num_classes)))

类别到数字id的转换只用了一行，妙啊，zip和dict还能这种配合

原网站

版权声明
本文为[氵文大师]所创，转载请带上原文链接，感谢
https://blog.csdn.net/HaoZiHuang/article/details/126241925

边栏推荐

猜你喜欢

随机推荐