def main(argv):
del argv
out_path = os.path.join(_output_dir(), 'cats_vs_dogs.zip')
jpg = _utils.get_random_jpeg(height=1, width=1)
with zipfile.ZipFile(out_path, 'w') as myzip:
myzip.write(jpg, 'PetImages/Dog/0.jpg')
myzip.write(jpg, 'PetImages/Dog/1.jpg')
myzip.write(jpg, 'PetImages/Cat/0.jpg')
myzip.write(jpg, 'PetImages/Cat/1.jpg')
def _generate_val_archive():
"""Generate val archive."""
output_path = os.path.join(_ilsvrc2012_output_dir(),
'ILSVRC2012_img_val.tar')
tar = tarfile.open(output_path, mode='w')
for i in range(1, VAL_IMAGES_NUMBER + 1):
fname = 'ILSVRC2012_val_0000%03i.JPEG' % i
jpeg = _utils.get_random_jpeg()
tar.add(jpeg, arcname=fname)
tar.close()
def _get_synset(synset_name):
"""Returns path to synset archive."""
fobj = tempfile.NamedTemporaryFile(delete=False, mode='wb', suffix='.tar')
tar = tarfile.open(mode='w', fileobj=fobj)
for i in range(1, TRAIN_IMAGES_PER_SYNSET + 1):
fname = '%s_%s.JPEG' % (synset_name, i)
jpeg = _utils.get_random_jpeg()
tar.add(jpeg, arcname=fname)
fobj.close()
return fobj.name
def _write_tar(path, split_name, image_ids, prefix=None):
"""Writes tar file with images to given path.
Args:
path: sting, path to tar to be written.
split_name: string. eg: 'train', 'validation', 'test'.
image_ids: list of str, ids of the images to add to tar file.
prefix: one of [0-9a-f], or None.
"""
if prefix is not None:
split_name = '%s_%s' % (split_name, prefix)
with tarfile.open(path, mode='w') as tar:
for image_id in image_ids:
fname = '%s/%s.jpg' % (split_name, image_id)
tar.add(_utils.get_random_jpeg(), arcname=fname)