Hi,
I am currently working on regression problem where the count of instances is required. I am using the Vgg16BN model. my code is as follows
train_data_dir = r'FPTest\train'
validation_data_dir = r'FPTest\valid'
trn=get_batches_t(train_data_dir)
val=get_batches_v(validation_data_dir)
model_v=Vgg16BN()
model_v.pop()
for layer in model_v.layers: layer.trainable=False
model_v.add(Dense(1,activation='Relu'))
however I get the following error.
C:\Users\satish\Anaconda\Deep Test>python fiber_count.py
Using Theano backend.
Using gpu device 0: GeForce 840M (CNMeM is enabled with initial size: 80.0% of memory, cuDNN 5103)
Found 123 images belonging to 10 classes.
Found 20 images belonging to 10 classes.
Traceback (most recent call last):
File "fiber_count.py", line 63, in <module>
model_v.pop()
AttributeError: Vgg16BN instance has no attribute 'pop'
not sure why I am getting this error. Anybody else face similar problem?
ps. have modified the get_batches so that the labels get converted into float with the number of instances.
Regards
Satish