Ciara
commited on
Commit
·
8c65296
1
Parent(s):
fdd61f4
Initial Files
Browse files- temporalnetv3.ckpt +3 -0
- temporalvideo.py +110 -0
temporalnetv3.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0a5b9f31f6d7e11aed40da84399c3aeebe7129a9788c0199cfa5d579d9363607
|
| 3 |
+
size 12688107909
|
temporalvideo.py
ADDED
|
@@ -0,0 +1,110 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import glob
|
| 3 |
+
import requests
|
| 4 |
+
import json
|
| 5 |
+
from pprint import pprint
|
| 6 |
+
import base64
|
| 7 |
+
from io import BytesIO
|
| 8 |
+
|
| 9 |
+
|
| 10 |
+
# Replace with the actual path to your image file and folder
|
| 11 |
+
x_path = "./init.png"
|
| 12 |
+
y_folder = "./New folder"
|
| 13 |
+
|
| 14 |
+
output_folder = "output"
|
| 15 |
+
os.makedirs(output_folder, exist_ok=True)
|
| 16 |
+
|
| 17 |
+
def get_image_paths(folder):
|
| 18 |
+
image_extensions = ("*.jpg", "*.jpeg", "*.png", "*.bmp")
|
| 19 |
+
files = []
|
| 20 |
+
for ext in image_extensions:
|
| 21 |
+
files.extend(glob.glob(os.path.join(folder, ext)))
|
| 22 |
+
return sorted(files)
|
| 23 |
+
|
| 24 |
+
y_paths = get_image_paths(y_folder)
|
| 25 |
+
|
| 26 |
+
def send_request(init_image_path, temp_path,controlnet_input_image_path):
|
| 27 |
+
url = "http://localhost:7860/sdapi/v1/img2img"
|
| 28 |
+
|
| 29 |
+
with open(init_image_path, "rb") as f:
|
| 30 |
+
init_image = base64.b64encode(f.read()).decode("utf-8")
|
| 31 |
+
|
| 32 |
+
with open(controlnet_input_image_path, "rb") as b:
|
| 33 |
+
controlnet_input_image = base64.b64encode(b.read()).decode("utf-8")
|
| 34 |
+
|
| 35 |
+
data = {
|
| 36 |
+
"init_images": [init_image],
|
| 37 |
+
"inpainting_fill": 0,
|
| 38 |
+
"inpaint_full_res": True,
|
| 39 |
+
"inpaint_full_res_padding": 1,
|
| 40 |
+
"inpainting_mask_invert": 1,
|
| 41 |
+
"resize_mode": 0,
|
| 42 |
+
"denoising_strength": 0.5,
|
| 43 |
+
"prompt": "anime man at computer",
|
| 44 |
+
"negative_prompt": "(ugly:1.3), (fused fingers), (too many fingers), (bad anatomy:1.5), (watermark:1.5), (words), letters, untracked eyes, asymmetric eyes, floating head, (logo:1.5), (bad hands:1.3), (mangled hands:1.2), (missing hands), (missing arms), backward hands, floating jewelry, unattached jewelry, floating head, doubled head, unattached head, doubled head, head in body, (misshapen body:1.1), (badly fitted headwear:1.2), floating arms, (too many arms:1.5), limbs fused with body, (facial blemish:1.5), badly fitted clothes, imperfect eyes, untracked eyes, crossed eyes, hair growing from clothes, partial faces, hair not attached to head",
|
| 45 |
+
"alwayson_scripts": {
|
| 46 |
+
"ControlNet":{
|
| 47 |
+
"args": [
|
| 48 |
+
{
|
| 49 |
+
"input_image": controlnet_input_image,
|
| 50 |
+
"module": "hed",
|
| 51 |
+
"model": "control_hed-fp16 [13fee50b]",
|
| 52 |
+
"weight": 1,
|
| 53 |
+
"guidance": 0.6,
|
| 54 |
+
},
|
| 55 |
+
{
|
| 56 |
+
"input_image": init_image,
|
| 57 |
+
"model": "temporalnetv3 [b146ac48]",
|
| 58 |
+
"module": "none",
|
| 59 |
+
"weight": 1,
|
| 60 |
+
"guidance": 0.8,
|
| 61 |
+
}
|
| 62 |
+
|
| 63 |
+
]
|
| 64 |
+
}
|
| 65 |
+
},
|
| 66 |
+
"seed": 770169050,
|
| 67 |
+
"subseed": -1,
|
| 68 |
+
"subseed_strength": -1,
|
| 69 |
+
"sampler_index": "Euler a",
|
| 70 |
+
"batch_size": 1,
|
| 71 |
+
"n_iter": 1,
|
| 72 |
+
"steps": 20,
|
| 73 |
+
"cfg_scale": 6,
|
| 74 |
+
"width": 1152,
|
| 75 |
+
"height": 640,
|
| 76 |
+
"restore_faces": True,
|
| 77 |
+
"include_init_images": True,
|
| 78 |
+
"override_settings": {},
|
| 79 |
+
"override_settings_restore_afterwards": True
|
| 80 |
+
}
|
| 81 |
+
response = requests.post(url, json=data)
|
| 82 |
+
if response.status_code == 200:
|
| 83 |
+
return response.content
|
| 84 |
+
else:
|
| 85 |
+
try:
|
| 86 |
+
error_data = response.json()
|
| 87 |
+
print("Error:")
|
| 88 |
+
print(str(error_data))
|
| 89 |
+
|
| 90 |
+
except json.JSONDecodeError:
|
| 91 |
+
print(f"Error: Unable to parse JSON error data.")
|
| 92 |
+
return None
|
| 93 |
+
|
| 94 |
+
output_images = []
|
| 95 |
+
output_images.append(send_request(x_path,y_folder, y_paths[0]))
|
| 96 |
+
output_paths = []
|
| 97 |
+
|
| 98 |
+
for i in range(1, len(y_paths)):
|
| 99 |
+
result_image = output_images[i-1]
|
| 100 |
+
temp_image_path = os.path.join(output_folder, f"temp_image_{i}.png")
|
| 101 |
+
data = json.loads(result_image)
|
| 102 |
+
encoded_image = data["images"][0]
|
| 103 |
+
with open(temp_image_path, "wb") as f:
|
| 104 |
+
f.write(base64.b64decode(encoded_image))
|
| 105 |
+
output_paths.append(temp_image_path)
|
| 106 |
+
result = send_request(y_paths[i], y_folder, temp_image_path)
|
| 107 |
+
output_images.append(result)
|
| 108 |
+
print(f"Written data for frame {i}:")
|
| 109 |
+
|
| 110 |
+
|