extractFrameAuto.py

#!/usr/bin/env python

#
#  make sure to run install.sh before trying this script
#  for exif data manipulation
#

import cv,cv2
from gi.repository import GExiv2
from fractions import Fraction
import argparse,re,time,os,sys
import random,math

parser = argparse.ArgumentParser(description='Program transforms video into seperate images for use in visual SFM')

parser.add_argument('-still', action="store",
    help='file to read exif from,must be of the same camera', 
    dest="still", default=None)

parser.add_argument('-file', action="store",
    help='file to transform', 
    dest="files", default=None)

parser.add_argument('-n', action="store",
    help='use only nth image',  
    dest="capture_step", default=1)

args = parser.parse_args()

capture_step = int(args.capture_step)
if capture_step < 1: capture_step = 1

tf = ''.join(args.files)
files = []
for f in tf.split(' '):
    files.append(f.split(','))

# N = F/D
# N = f number
# F = focal lenght
# D = diameter of lens

files = [item for sublist in files for item in sublist]
print "Converting files:",files

exif1 = GExiv2.Metadata(args.still)

print "Using file:%s as exif data input"%args.still
print "values:"
print "FNumber:",exif1['Exif.Photo.FNumber']
print "Focal Length:",exif1['Exif.Photo.FocalLength']
print "Aperture Value:",exif1['Exif.Photo.ApertureValue']
print "Camera Model:",exif1['Exif.Image.Model']
print "Camera Brand:",exif1['Exif.Image.Make']

for f in files:
    capture = cv2.VideoCapture(f)

    width = capture.get(cv.CV_CAP_PROP_FRAME_WIDTH)
    height = capture.get(cv.CV_CAP_PROP_FRAME_HEIGHT)
    fps = capture.get(cv.CV_CAP_PROP_FPS)
    frame_count =  capture.get(cv.CV_CAP_PROP_FRAME_COUNT)
    codec = capture.get(cv.CV_CAP_PROP_FOURCC)

    print "Starting work on %s now" % f

    if capture_step > frame_count: frame_count = frame_count - 1

    ISOSPEEDS = [64, 100, 200, 250, 320, 400, 640, 800, 1000, 1600, 3200]

    SHUTTERSPEEDS = [15, 30, 60, 125, 250, 400, 
    500, 1000, 1250, 1600, 2000, 4000]
    
    FSTOPS = [1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.7, 
    1.8, 2, 2.2, 2.4, 2.6, 2.8, 3.2, 3.4, 3.7, 
    4, 4.4, 4.8, 5.2, 5.6, 6.2, 6.7, 7.3, 8, 8.7, 
    9.5, 10, 11, 12, 14, 15, 16, 17, 19, 21, 22]
    
    flash = [0x00,#No flash
    0x1, 
    0x18, 
    0x19, 
    0x49, 
    0x4d, 
    0x4f, 
    0x49, 
    0x4d, 
    0x4f]

    aperture = Fraction(random.uniform(1.0, 16.0)).limit_denominator(2000)
    exposure = Fraction(1.0/round(random.randint(8, int(100.0*aperture))+1, -2)).limit_denominator(4000)
    
    for i in xrange(int(frame_count)):
        ret, frame = capture.read()
        if ret and (i % capture_step == 0):
            sys.stdout.write('saving frame:%s\r'%i)
            sys.stdout.flush()
            path = "%s.jpg"%(i)
            cv2.imwrite(path, frame, [int(cv2.IMWRITE_JPEG_QUALITY), 90])

           
            exif = GExiv2.Metadata(path)
            
            t = os.path.getctime(path)
            ctime = time.strftime('%d/%m/%Y %H:%M:%S', time.localtime(t))

            exif['Exif.Image.ImageDescription'] = "SEQ#%s"%i
            exif['Exif.Image.Make'] = exif1['Exif.Image.Make']
            exif['Exif.Image.Model'] = exif1['Exif.Image.Model']
            exif['Exif.Image.DateTime'] = ctime
            exif['Exif.Image.Software'] = "https://github.com/eokeeffe/videoExtractor"
            exif['Exif.Image.Orientation'] = exif1['Exif.Image.Orientation']

            exif['Exif.Photo.UserComment'] = "awesomeness"
            exif['Exif.Photo.Flash'] = exif1['Exif.Photo.Flash']
            exif['Exif.Photo.FNumber'] = exif1['Exif.Photo.FNumber']
            exif['Exif.Photo.FocalLength'] = exif1['Exif.Photo.FocalLength']
            exif['Exif.Photo.ApertureValue'] = exif1['Exif.Photo.ApertureValue']
            exif['Exif.Photo.ExposureTime'] = exif1['Exif.Photo.ExposureTime']
            exif['Exif.Photo.ExposureBiasValue'] = exif1['Exif.Photo.ExposureBiasValue']
            exif['Exif.Photo.ISOSpeedRatings'] = exif1['Exif.Photo.ISOSpeedRatings']

            exif.save_file()