def vid_preprocess(args):
base_dir = args.base_dir
out_dir = args.out_dir
mix_dir = set_dir(base_dir)
fps = args.fps
vid_size = (args.vid_w, args.vid_h)
mix_clip = mix_vid_clip_read(mix_dir, fps)
mix_clip_frames = int(mix_clip.fps * mix_clip.duration)
# show video frame
for frame in range(mix_clip_frames):
# first column
vid0 = frame_extract(mix_clip, frame, None, fps, vid_size, BGR2RGB=True)
cv2.imwrite(join(out_dir, 'mix_' + str(frame) + '.png'), vid0)
print(frame)
def vid_labeling(args):
base_dir = args.base_dir
mix_dir, label_dir, vid_dir = set_dir(base_dir)
vid_size = (args.vid_w, args.vid_h)
fps = args.fps
# CSV file writer
CSVlogger = LogCSV(log_dir=join(base_dir, "label.csv"))
mix_clip = mix_vid_clip_read(mix_dir, fps)
mix_clip_frames = int(mix_clip.fps * mix_clip.duration)
vid_clip = origin_vid_clip_read(vid_dir, mix_clip_frames, fps)
# show video video frames in window
for frame in range(mix_clip_frames):
# first column
vid0 = frame_extract(vid_clip, frame, 0, fps, vid_size)
vid1 = frame_extract(vid_clip, frame, 1, fps, vid_size)
vid2 = frame_extract(vid_clip, frame, 2, fps, vid_size)
img1 = np.concatenate((vid0, vid1, vid2), axis=0)
# second column
vid3 = frame_extract(vid_clip, frame, 3, fps, vid_size)
mix = frame_extract(mix_clip, frame, None, fps, vid_size)
vid4 = frame_extract(vid_clip, frame, 4, fps, vid_size)
img2 = np.concatenate((vid3, mix, vid4), axis=0)
# third column
vid5 = frame_extract(vid_clip, frame, 5, fps, vid_size)
vid6 = frame_extract(vid_clip, frame, 6, fps, vid_size)
vid7 = frame_extract(vid_clip, frame, 7, fps, vid_size)
img3 = np.concatenate((vid5, vid6, vid7), axis=0)
# concat all column
img = np.concatenate((img1, img2, img3), axis=1)
cv2.imshow("", img)
k = int(cv2.waitKey(0))
# 0 is 48 in window
# below value is ubuntu setting, if you use windows system, please delete below operation
k -= 48
print(k)
CSVlogger([frame, k])