Transfer learning with tensorflow

[Ajaz-Ahmad]

Hi,

I am using yolo architecture and trying transfer learning. I have pre-trained weights trained over 80 classes but now I want output for 3 classes only. How can I remove the last convolution layer and fit my own layer?
Below is the initializer method where I am restoring model and training only the last part but I want to change last convolutional layer as well.
Thanks