def flowLayer(self, net, **kwargs):
kwargs['setting'] = 'OPTICAL_FLOW_SINGLE'
kwargs['rendertype'] = self._rendertype
kwargs['phase'] = self._phase
kwargs['collection_list'] = COLL_LISTS_DIR + '/sintel_train.txt'
kwargs['rand_permute'] = False
return Data.BinaryData(net, **kwargs)
def sceneFlowLayer(self, net, **kwargs):
kwargs['setting'] = 'SCENE_FLOW_SINGLE'
kwargs['rendertype'] = self._rendertype
kwargs['phase'] = self._phase
kwargs['collection_list'] = COLL_LISTS_DIR + '/v1/monkaa_release.txt'
kwargs['rand_permute'] = False
return Data.BinaryData(net, **kwargs)
def dispLayer(self, net, **kwargs):
kwargs['setting'] = 'DISPARITY_SINGLE'
kwargs['rendertype'] = self._rendertype
kwargs['phase'] = self._phase
kwargs['collection_list'] = COLL_LISTS_DIR + '/kitti2015_train.txt'
kwargs['rand_permute'] = False
return Data.BinaryData(net, **kwargs)
def flowLayer(self, net, **kwargs):
kwargs['setting'] = 'OPTICAL_FLOW_SINGLE'
kwargs['rendertype'] = self._rendertype
kwargs['phase'] = self._phase
kwargs[
'collection_list'] = COLL_LISTS_DIR + '/v1/FlyingStuff3D_new_WebP_test.txt' #FlyingStuff3D_new_test.txt'
kwargs['rand_permute'] = False
return Data.BinaryData(net, **kwargs)
def sceneFlowLayer(self, net, **kwargs):
kwargs['setting'] = 'SCENE_FLOW_SINGLE_REDUCED'
kwargs['rendertype'] = self._rendertype
kwargs['phase'] = self._phase
kwargs['collection_list'] = COLL_LISTS_DIR + '/kitti2015_train.txt'
kwargs['rand_permute'] = False
blobs = Data.BinaryData(net, **kwargs)
batch_size = kwargs['batch_size'] if 'batch_size' in kwargs else 1
blobs.gt.flowR = net.zeros(batch_size, 2, self.height(), self.width())
blobs.gt.disp1L = net.zeros(batch_size, 1, self.height(), self.width())
return blobs