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

最新推荐文章于 2024-09-15 22:31:42 发布

氵文大师

最新推荐文章于 2024-09-15 22:31:42 发布

阅读量166

点赞数

分类专栏：每日一氵文章标签： python

本文链接：https://blog.csdn.net/HaoZiHuang/article/details/126241925

版权

每日一氵专栏收录该内容

162 篇文章 13 订阅

订阅专栏

在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)))