コード例 #1
0
        help="maximum of frames to sample from one video",
        type=int,
        default=30
    )
    args = parser.parse_args()

    del_upper = 100
    FRAMERATE = args.fr
    OUT_DIR = args.out
    MAX_FRAMES = args.max
    print("READ DATA FROM ", args.data)
    print("OUTPUT DATA IN ", args.out)
    print("SELECT FRAMES AT", args.fr, "Hz")
    print("SELECT MAXIMUM OF", args.max, "frames")

    actual_names, labels = get_paths(args.json)
    # manually add the ones which I know are in the data
    files_to_process, labs_to_process = get_processing_info(
        args.data, actual_names, labels
    )
    makedirs(OUT_DIR)

    # RUN - Iterate over videos
    for i in range(len(files_to_process)):  # 1,2): #
        fp = files_to_process[i]
        fn = fp.split(os.sep)[-1]
        cap = cv2.VideoCapture(fp)  # capturing the video from the given path
        # frame rate
        n_frames = cap.get(7)
        frameRate = cap.get(5)
        every_x_image = int(frameRate / FRAMERATE)
コード例 #2
0
    parser.add_argument('-max',
                        help="maximum of frames to sample from one video",
                        type=int,
                        default=30)
    args = parser.parse_args()

    del_upper = 100
    FRAMERATE = args.fr
    OUT_DIR = args.out
    MAX_FRAMES = args.max
    print("READ DATA FROM ", args.data)
    print("OUTPUT DATA IN ", args.out)
    print("SELECT FRAMES AT", args.fr, "Hz")
    print("SELECT MAXIMUM OF", args.max, "frames")

    actual_names, labels = get_paths(args.csv)
    # manually add the ones which I know are in the data
    files_to_process, labs_to_process = get_processing_info(
        args.data, actual_names, labels)
    makedirs(OUT_DIR)

    # RUN - Iterate over videos
    for i in range(len(files_to_process)):  # 1,2): #
        fp = files_to_process[i]
        fn = fp.split(os.sep)[-1]
        cap = cv2.VideoCapture(fp)  # capturing the video from the given path
        # frame rate
        n_frames = cap.get(7)
        frameRate = cap.get(5)
        every_x_image = int(frameRate / FRAMERATE)
        out_path = os.path.join(OUT_DIR, label_to_dir(labs_to_process[i]))