Retraining the captioning model

So, now that we have seen image captioning code in action, we are going to retrain the image captioner on our own desired data. However, we need to know that it will be very time consuming and will need over 100 GB of hard drive space for computations if we want it to process in a reasonable time. Even with a good GPU, it may take a few days or a week to complete the computation. Since we are inclined toward implementing it and have the resources, let's start retraining the model.

In the Notebook, the first step is to download the pre-trained Inception model. The webbrowser module will make it easy to open the URL and to download the file:

# First download pretrained Inception (v3) modelimport webbrowser 

Get Computer Vision Projects with OpenCV and Python 3 now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.