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
评论列表
文章目录