inception_builder.py 文件源码

def build(self, network_filename='inception.prototxt'):
        """main method."""

        network = self._build_network()
        print network
        with open(network_filename, 'w') as network_file:
            network_file.write(text_format.MessageToString(network))
        return Net(network_filename)

#if __name__ == '__main__':
    #__inception_builder__ = InceptionBuilder(
        #training_source='examples/imagenet/ilsvrc12_train_lmdb',
        #training_batch_size=32,
        #testing_source='examples/imagenet/ilsvrc12_val_lmdb',
        #testing_batch_size=6,
        #task_name='ilsvrc12',
        #mean_file='data/ilsvrc12/imagenet_mean.binaryproto',
        #labels=[('class_id', 1000)],
    #)

    #__inception_builder__.build()