nyu_rgbd.py 文件源码

def iteritems(self, every_k_frames=1): 
        index = 0 
        # , bbox, pose
        bbox, pose = None, None
        for rgb_im, depth_im, instance, label in izip(islice(self._ims, 0, None, every_k_frames), 
                                                      islice(self._depths, 0, None, every_k_frames), 
                                                      islice(self._instances, 0, None, every_k_frames), 
                                                      islice(self._labels, 0, None, every_k_frames)
        ): 
            index += every_k_frames
            yield self._process_items(index, rgb_im, depth_im, instance, label, bbox, pose)

    # def iterinds(self, inds): 
    #     for index, rgb_im, depth_im, bbox, pose in izip(inds, 
    #                                                     self.rgb.iterinds(inds), 
    #                                                     self.depth.iterinds(inds), 
    #                                                     [self.bboxes[ind] for ind in inds], 
    #                                                     [self.poses[ind] for ind in inds]): 
    #         yield self._process_items(index, rgb_im, depth_im, bbox, pose)