remove start/stop

This commit is contained in:
rlx 2021-08-09 15:45:30 +00:00
parent 2f6f44ec96
commit 5f0281f14f

4
pc.py
View file

@ -38,7 +38,6 @@ cell_w, cell_h = w//m, h//m
c = 1 # Color channels c = 1 # Color channels
fps = 24 fps = 24
kf = 1 # Keyframes (every kf-th source frame is a cell frame candidate) kf = 1 # Keyframes (every kf-th source frame is a cell frame candidate)
start, stop = 3275, 97701
ts, ti = 1, 0 # Number of threads, current thread index ts, ti = 1, 0 # Number of threads, current thread index
cache = {} cache = {}
@ -69,9 +68,8 @@ def load_image(fn):
def read_images(): def read_images():
def save_data(): def save_data():
np.save(data_fn, data) np.save(data_fn, data)
src_fn = f'{src_dn}/{stop:08d}.jpg'
if not os.path.exists(src_dn): if not os.path.exists(src_dn):
os.makedirs(src_dn, exist_ok=True) os.makedirs(src_dn)
os.system(f'ffmpeg -i "{video_fn}" -q:v 1 "{src_dn}/%08d.jpg"') os.system(f'ffmpeg -i "{video_fn}" -q:v 1 "{src_dn}/%08d.jpg"')
cell_dn = f'{src_dn}_{cell_w}x{cell_h}' cell_dn = f'{src_dn}_{cell_w}x{cell_h}'
os.makedirs(cell_dn, exist_ok=True) os.makedirs(cell_dn, exist_ok=True)