# --------------------------------------------------------# Caffe for Dragon# Copyright(c) 2017 SeetaTech# Written by Ting Pan# --------------------------------------------------------importdragon.vm.caffeascaffeimportdragon.core.workspaceaswsfromminibatchimportDataBatchclassDataLayer(caffe.Layer):defsetup(self,bottom,top):kwargs=eval(self.param_str)self._data_batch=DataBatch(**kwargs)defforward(self,bottom,top):blobs=self._data_batch.blobsforidx,blobinenumerate(blobs):ws.FeedTensor(top[idx],blob)