Computer Vision News - January 2022
15 Drawing Neural Networks Pytorch model Keras model from torchvision import models from torchsummary import summary vgg = models.vgg16() summary(vgg, ( 3 , 224 , 224 )) import keras from keras.applications import vgg16 vgg = vgg16.VGG16(weights="imagenet") print (vgg.summary())
Made with FlippingBook
RkJQdWJsaXNoZXIy NTc3NzU=