vgg16.py 文件源码

def __init__(self, vgg16_npy_path=None, trainable=True):
        if vgg16_npy_path is None:
            path = inspect.getfile(VGG16)
            path = os.path.abspath(os.path.join(path, os.pardir))
            path = os.path.join(path, "vgg16.npy")
            vgg16_npy_path = path
            print(path)
        self.trainable = trainable
        self.imgs = tf.placeholder(tf.float32, [None, 224, 224, 3], name='images')
        self.labels = tf.placeholder(tf.float32, [None, FLAGS.num_classes], name='labels')
        self.var_dict = {}
        self.temp_value = None
        self.data_dict = None
        with open(vgg16_npy_path, 'rb') as f:
            self.data_dict = np.load(vgg16_npy_path, encoding='latin1').item()
        # # self.data_dict = pickle.load(f)
        #    pickle.dump(self.data_dict, f, protocol=2) 
        self.lrn_rate = 0.01
        print("npy file loaded")
        self.build()
        self.loss_layer()