How to export non txt data from Colaboratory to Google Drive?

Question

I'm running Colab and I want to save some non-txt data (numpy arrays, PIL images, .h5 keras/tensorflow models) to my drive.

I'm able to save .txt files using this script

from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive
from google.colab import auth
from oauth2client.client import GoogleCredentials

# 1. Authenticate and create the PyDrive client.
auth.authenticate_user()
gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.get_application_default() 
drive = GoogleDrive(gauth)

# PyDrive reference:
# https://googledrive.github.io/PyDrive/docs/build/html/index.html

# 2. Create & upload a file text file.
uploaded = drive.CreateFile({'title': 'Sample upload.txt'})
uploaded.SetContentString('Sample upload file content')
uploaded.Upload()
print('Uploaded file with ID {}'.format(uploaded.get('id')))

# 3. Load a file by ID and print its contents.
downloaded = drive.CreateFile({'id': uploaded.get('id')})
print('Downloaded content "{}"'.format(downloaded.GetContentString()))

but I can't using it for other types of data.

Any help would be very much appreciated!

Answer 1

pydrive supports uploading a file as well as a string -- see this sample in the docs.

In addition, you can also set the MIME type when creating the file, eg

uploaded = drive.CreateFile({'title': 'sample.csv', 'mimeType': 'text/csv'})