Tensorflow 에서 VGG16을 사용하는 방법

TF 에서 vgg16을 빌드하는 방법

• https://github.com/tensorflow/models/tree/master/research/slim 에서 vgg16의 checkpoint file을 다운로드 받는다.

• vgg16 graph 를 coding 한다. (https://github.com/penny4860/cnn-visualizer/blob/master/src/vgg.py)
• naming scope를 ckpt와 맞춘다.
• 다운로드받은 checkpoint 로부터 value를 restore 한다.

        variables = slim.get_variables()    
        init_assign_op, init_feed_dict = slim.assign_from_checkpoint('ckpts/vgg_16.ckpt', variables)
        with tf.Session() as sess:
            sess.run(init_assign_op, init_feed_dict)
            filter_ = sess.run(variables[0])
            value = filter_[:,:,:,0].reshape(-1,)
            print(value)