Stack Overflow
Stack Overflow

Questions tagged [vgg-net]

A kind of convolutional neural network consisting of 16 or 19 layers, often used with weights pre-trained on ImageNet dataset. Whereas the the model was originally created for image classification, its convolutional part can be used for a variety of purposes. Use this tag for questions, specific for this CNN architecture.

The name VGG stands for Visual Geometry Group (Oxford University), authors of the original paper.

The model consists of a convolutional part (several convolution and max- or avegare-pooling layers) and several fully-connected layers atop of it. Small (3x3) convolution filters are used.

See visual representation below (taken from this answer):

enter image description here

Model applications

  1. Image classifier (Tensorflow).
  2. Image segmentation (Keras).
  3. Image style transfer (Keras).
471 questions
-1
votes
1 answer

VGG Transfer learning - predict generator shows different result than evaluate generator

I built a CNN based on VGG pre-trained model. I fine tuned the last conv block (#5) together with my added Fully connected layers. My classification problem has 4 classes, my last activation layer is 'softmax' and I am using…
Hamid K
  • 983
  • 1
  • 18
  • 40
-1
votes
1 answer

Saving the weights of VGG-16 model trained in keras

How to save the weights of VGG-16 model after training it? How to load the saved weights in to the model? I tried this: fname =…
A Santosh
  • 839
  • 1
  • 9
  • 11
-2
votes
1 answer

VGG13 pretrained ImageNet weights

i'm trying to implement the VGG13 model in keras but i'm having a lot of difficulties in finding the ImageNet pretrained weights for it. Does anyone know where to find does?
Elvopresla
  • 147
  • 2
  • 14
-2
votes
2 answers

Target size (torch.Size([12])) must be the same as input size (torch.Size([12, 1000]))

I am using models.vgg16(pretrained=True) model for image classification, where number of classes = 3. Batch size is 12 trainloader = torch.utils.data.DataLoader(train_data, batch_size=12, shuffle=True) since error says Target size (torch.Size([12]))…
Irfan Umar
  • 196
  • 3
  • 18
-2
votes
2 answers

Feature extraction with VGGnet from my own data with TensorFlow?

I pre-trained VGG19 with SIRI-WHU dataset, and now I want to extract the features, and I don't know how to do it. Can someone help me please? thanks
hanen balti
  • 31
  • 5
-3
votes
1 answer

tensorFlow Vgg16 algorithm training on different dataset

I'm trying to train a VGG16 model following a video guide on YouTube. I copied the code given by the instructor. After doing this I tried to train a model using some of the images available in my system. I have uploaded some images only to…
Mohammad Awais
  • 35
  • 6
1 2 3
…
31
32