deepgeek/MuseTalk
2
0
Fork 0
mirror of https://github.com/TMElyralab/MuseTalk.git synced 2026-02-05 18:09:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #197 from ShowLo/patch-1

Browse Source 
fix bug in preprocessing data
This commit is contained in:
czk32611
2024-09-23 13:27:15 +08:00
committed by GitHub
parent 98f0e6f2b1 5bd772d7da
commit b3c30c3561
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/data.py
View File

@@ -183,12 +183,13 @@ def main(args):
video_num = len(whisper_chunks) video_num = len(whisper_chunks)
batch_size = args.batch_size batch_size = args.batch_size
gen = datagen(whisper_chunks,crop_data,batch_size) gen = datagen(whisper_chunks,crop_data,batch_size)
crop_index = 0
for i, (whisper_batch,crop_batch) in enumerate(tqdm(gen,total=int(np.ceil(float(video_num)/batch_size)))): for i, (whisper_batch,crop_batch) in enumerate(tqdm(gen,total=int(np.ceil(float(video_num)/batch_size)))):
crop_index=0
for image,audio in zip(crop_batch,whisper_batch): for image,audio in zip(crop_batch,whisper_batch):
cv2.imwrite(f"data/images/{folder_name}/{str(i+crop_index+total_image_index+1)}.png",image) cv2.imwrite(f"data/images/{folder_name}/{str(crop_index+total_image_index+1)}.png",image)
crop_index+=1 temp_image_index = crop_index + total_image_index + 1
temp_image_index=i+crop_index+total_image_index+1 crop_index += 1
# np.save(f'data/audios/{folder_name}/{str(i+crop_index)}.npy', audio) # np.save(f'data/audios/{folder_name}/{str(i+crop_index)}.npy', audio)
total_image_index=temp_image_index total_image_index=temp_image_index
gc.collect() gc.collect()