return ground truth image regions database :return: imdb[image_index]['boxes', 'gt_classes', 'gt_overlaps', 'flipped']
(self)
| 90 | return seg_class_file |
| 91 | |
| 92 | def gt_roidb(self): |
| 93 | """ |
| 94 | return ground truth image regions database |
| 95 | :return: imdb[image_index]['boxes', 'gt_classes', 'gt_overlaps', 'flipped'] |
| 96 | """ |
| 97 | cache_file = os.path.join(self.cache_path, self.name + '_gt_roidb.pkl') |
| 98 | if os.path.exists(cache_file): |
| 99 | with open(cache_file, 'rb') as fid: |
| 100 | roidb = cPickle.load(fid) |
| 101 | print '{} gt roidb loaded from {}'.format(self.name, cache_file) |
| 102 | return roidb |
| 103 | |
| 104 | gt_roidb = [self.load_pascal_annotation(index) for index in self.image_set_index] |
| 105 | with open(cache_file, 'wb') as fid: |
| 106 | cPickle.dump(gt_roidb, fid, cPickle.HIGHEST_PROTOCOL) |
| 107 | print 'wrote gt roidb to {}'.format(cache_file) |
| 108 | |
| 109 | return gt_roidb |
| 110 | |
| 111 | def gt_segdb(self): |
| 112 | """ |
nothing calls this directly
no test coverage detected