VGG16模型
import tensorflow as tfclass Vgg16(): def __init__(self,imgs): self.parameters = [] self.imgs = [] self.convlayers() self.fc_layers() def conv(self,name,input_data,out_channel,trainable=False): in_channel = input_data.get_shape()[-1] with
原创
2020-10-21 22:05:16 ·
206 阅读 ·
0 评论