import numpy as np
import cv2
import sys, os, time
import numpy as np
import simplejson
import sys, os
import csv
CURR_PATH = os.path.dirname(os.path.abspath(__file__)) + "/"
sys.path.append(CURR_PATH + "../")
from mylib.funcs import get_filenames
CURR_PATH = os.path.dirname(os.path.abspath(__file__)) + "/"

if 1:
    image_folder = CURR_PATH + '../skeleton_data/skeletons5apple_images/'
    video_name = CURR_PATH + '../mytest.avi'
    fnames = get_filenames(image_folder)
    N = len(fnames)
    image_start = 3
    image_end = 200
    framerate = 10
    FASTER_RATE = 1

# Read image and save to video'
fourcc = cv2.VideoWriter_fourcc(*'XVID')
cnt = 0
for i in range(image_start, image_end + 1):
    cnt += 1
    fname = "{:05d}.png".format(i)
    frame = cv2.imread(image_folder + fname)
    if cnt == 1:
        width = frame.shape[1]
 def __init__(self, folder, num_skip=0):
     self.cnt_image = 0
     self.folder = folder
     self.filenames = myfunc.get_filenames(folder, sort=True)
     self.idx_step = num_skip + 1
     self.num_images = int(len(self.filenames) / self.idx_step)
Example #3
0
import csv
CURR_PATH = os.path.dirname(os.path.abspath(__file__))+"/"
sys.path.append(CURR_PATH+"../")
from mylib.io import load_skeletons
from mylib.funcs import get_filenames
CURR_PATH = os.path.dirname(os.path.abspath(__file__))+"/"


if __name__=="__main__":
    # Parameters -------------------------------------------

    data_idx = "5"

    read_from = "../skeleton_data/skeletons"+data_idx+"/"
    output_to = "../skeleton_data/skeletons"+data_idx+"_info.txt"
    NUM_SKELETONS = len(get_filenames(CURR_PATH + read_from))

    # Main -------------------------------------------

    int2str = lambda num, blank: ("{:0"+str(blank)+"d}").format(num)
    all_skeletons = []
    all_skeletons_good_only = []
    for i in range(1, NUM_SKELETONS+1):
        if i%100==0:
            print("{}/{}".format(i, NUM_SKELETONS))
        skeletons = load_skeletons(CURR_PATH + read_from + int2str(i, 5) + ".txt")
        idx_person = 0 # Only one person in each image
        skeleton = skeletons[idx_person]

        # -- Push to result list
        all_skeletons.append(skeleton)