visual_genome_loader.py 文件源码

python
阅读 18 收藏 0 点赞 0 评论 0

项目:textobjdetection 作者: andfoy 项目源码 文件源码
def __filter_regions_by_class(self, regions):
        print("Filtering regions...")
        act_regions = []
        region_sub = {}
        bar = progressbar.ProgressBar()
        for region in bar(regions):
            try:
                reg_obj = self.region_objects[region.image.id][region.id]
                reg_obj = frozenset([x.lower()
                                     for x in reg_obj])
            except KeyError:
                reg_obj = frozenset({})

            if reg_obj in self.obj_idx:
                act_regions.append(region)
                if region.image.id not in region_sub:
                    region_sub[region.image.id] = {}
                reg_img = region_sub[region.image.id]
                global_region_img = self.region_objects[region.image.id]
                reg_img[region.id] = global_region_img[region.id]
        return act_regions, region_sub
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号