def extract(videoName, outputBase, dataset):
    """
    Extracts the IDTFs and stores them in outputBase file.
    """
    if not dataset == "Something":
        videoName = videoName[:-4]
        if not os.path.exists(videoName):
            print '%s does not exist!' % videoName
            return False
        print(videoName)
        print(os.path.basename(videoName))
        resizedName = os.path.join(
            tmpDir,
            os.path.basename(videoName) + '/image_%05d.jpg')
        if not ffmpeg.resize(videoName, resizedName):
            resizedName = videoName + '/image_%05d.jpg'  # resize failed, just use the input video
    else:
        if not os.path.exists(videoName):
            print '%s does not exist!' % videoName
            return False
        print(videoName)
        print(os.path.basename(videoName))
        resizedName = os.path.join(tmpDir,
                                   os.path.basename(videoName) + '/%05d.jpg')
        if not ffmpeg.resize(videoName, resizedName):
            resizedName = videoName + '/%05d.jpg'  # resize failed, just use the input video

    subprocess.call('%s %s > %s' % (dtBin, resizedName, outputBase),
                    shell=True)
    return True
def extract(videoName, outputBase):
    """
    Extracts the IDTFs and stores them in outputBase file.
    """
    if not os.path.exists(videoName):
        print '%s does not exist!' % videoName
        return False
    resizedName = os.path.join(tmpDir, os.path.basename(videoName))
    if not ffmpeg.resize(videoName, resizedName):
        resizedName = videoName     # resize failed, just use the input video
    subprocess.call('%s %s > %s' % (dtBin, resizedName, outputBase), shell=True)
    return True
Exemple #3
0
def extract(videoName, outputBase):
    if not os.path.exists(videoName):
        print '%s does not exist!' % videoName
        return False
    if check_dup(outputBase):
        print '%s processed' % videoName
        return True
    resizedName = os.path.join(tmpDir, os.path.basename(videoName))
    if not ffmpeg.resize(videoName, resizedName):
        resizedName = videoName     # resize failed, just use the input video
    subprocess.call('%s %s | %s %s %s %s %s' % (dtBin, resizedName, tevBin, projMatList, rMeanList, codeBookList, outputBase), shell=True)
    return True
Exemple #4
0
def extract(videoName, outputBase):
    """
    Extracts the IDTFs and stores them in outputBase file.
    """
    if not os.path.exists(videoName):
        print '%s does not exist!' % videoName
        return False
    resizedName = os.path.join(tmpDir, os.path.basename(videoName))
    if not ffmpeg.resize(videoName, resizedName):
        resizedName = videoName  # resize failed, just use the input video
    subprocess.call('%s %s > %s' % (dtBin, resizedName, outputBase),
                    shell=True)
    return True
def extractFV(videoName, outputBase, gmm_list, dataset):
    """
    Extracts the IDTFs, constructs a Fisher Vector, and saves the Fisher Vector at outputBase
    outputBase: the full path to the newly constructed fisher vector.
    gmm_list: file of the saved list of gmms
    """
    if not dataset == "Something":
        videoName = videoName[:-4]
        if not os.path.exists(videoName):
            print '%s does not exist!' % videoName
            return False
        resizedName = os.path.join(
            tmpDir,
            os.path.basename(videoName) + '/image_%05d.jpg')
        resized_vids = [filename for filename in os.listdir(tmpDir)]
        # resized_vids = [filename for filename in os.listdir(tmpDir) if filename.endswith('.avi')]
        if os.path.basename(videoName) not in resized_vids:
            if not ffmpeg.resize(videoName, resizedName):
                resizedName = videoName + '/image_%05d.jpg'  # resize failed, just use the input video
    else:
        if not os.path.exists(videoName):
            print '%s does not exist!' % videoName
            return False
        resizedName = os.path.join(tmpDir,
                                   os.path.basename(videoName) + '/%05d.jpg')
        resized_vids = [filename for filename in os.listdir(tmpDir)]
        # resized_vids = [filename for filename in os.listdir(tmpDir) if filename.endswith('.avi')]
        if os.path.basename(videoName) not in resized_vids:
            if not ffmpeg.resize(videoName, resizedName):
                resizedName = videoName + '/%05d.jpg'  # resize failed, just use the input video

    print videoName
    subprocess.call('%s %s | %s %s %s' %
                    (dtBin, resizedName, COMPUTE_FV, outputBase, gmm_list),
                    shell=True)
    return True
def extractFV(videoName, outputBase,gmm_list):
    """
    Extracts the IDTFs, constructs a Fisher Vector, and saves the Fisher Vector at outputBase
    outputBase: the full path to the newly constructed fisher vector.
    gmm_list: file of the saved list of gmms
    """
    if not os.path.exists(videoName):
        print '%s does not exist!' % videoName
        return False
    resizedName = os.path.join(tmpDir, os.path.basename(videoName))
    resized_vids = [filename for filename in os.listdir(tmpDir) if filename.endswith('.avi')]
    if os.path.basename(videoName) not in resized_vids:
        if not ffmpeg.resize(videoName, resizedName):
            resizedName = videoName     # resize failed, just use the input video
    print videoName
    subprocess.call('%s %s | %s %s %s' % (dtBin, resizedName, COMPUTE_FV, outputBase, gmm_list), shell=True)
    return True
                        help="Name of the input desired to test prediction",
                        type=str)
    parser.add_argument(
        '--no_pca',
        dest='no_pca',
        action='store_const',
        const=True,
        default=False,
        help='Testing without PCA reduction (default: uses PCA)')
    args = parser.parse_args()
    video = args.video_name

    vid = os.path.join(video_dir, video)
    if not check_resolution(vid):
        resizedName = os.path.join(tmp_dir, video)
        if ffmpeg.resize(vid, resizedName):
            vid = resizedName
    command = dtBin + ' ' + vid
    p = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE)
    raw_features, _ = p.communicate()
    features = raw_features.split("\n")
    points = []
    for point in features:
        if point != '' and point[
                0] != '[':  #to delete info messages from DenseTrackStab
            points.append(IDT_feature.IDTFeature(point))
    video_desc = IDT_feature.vid_descriptors(points)
    gmm_list = np.load(gmm_list + ".npz")['gmm_list']
    fish = computeFV.create_fisher_vector_unsaved(gmm_list, video_desc)
    if not args.no_pca:
        pca = classify_library.load_model('../data/models/pca.sav')