Scavengers Reign Style - LoRA

이 모델은 MAX에서 방영된 TV 시리즈 'Scavengers Reign'(2023)의 1000개의 무작위 스크린샷으로 학습되었습니다. 학습에는 CivitAI의 기본 설정을 사용했습니다.

워크플로우

1. MP4 파일에서 무작위 이미지를 추출하기 위해 짧은 Python 스크립트를 사용했습니다.

2. 그 후, czkawka(github)를 사용하여 중복되거나 유사한 이미지를 제거했습니다.

3. 그 다음, 모든 이미지를 수동으로 확인했습니다.

4. 캡셔닝에는 kohya_ss(github) 내의 WD14 모델을 사용했습니다.

5. 마지막으로 캡션과 이미지를 CivitAI에 업로드하여 학습을 진행했습니다.

코드:

import cv2
import random

mp4_directory = ''
output_directory = ''
frames_to_extract = 120
base_name = "Random_screenshot"
list_of_random_frames = []
frame_distance = 100
first_frame = 0 

count = 0

vidcap = cv2.VideoCapture(mp4_directory)
totalFrames = vidcap.get(cv2.CAP_PROP_FRAME_COUNT)
while count < frames_to_extract:
    count += 1
    count_str = str(count)
    frames_skipped = -1
    while True:
        randomFrameNumber = random.randint(0, totalFrames)
        frames_skipped +=1
        if frames_skipped > 0:
            print(f"Frame Skipped {frames_skipped}")
        if all(abs(randomFrameNumber - frame) > frame_distance and randomFrameNumber> first_frame for frame in list_of_random_frames):
            break
    list_of_random_frames.append(randomFrameNumber)
    photo_output = output_directory + basename + count_str + ".png"
    vidcap.set(cv2.CAP_PROP_POS_FRAMES,randomFrameNumber)
    success, image = vidcap.read()
    if success:
        cv2.imwrite(photo_output, image)
    print(f"Saving image to: {photo_output}")

참고: 데이터셋을 원하시면 저에게 연락해주세요. 단, CivitAI가 저작권 문제에 휘말리지 않도록 하기 위함입니다.