Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
C
ComfyUI-MimicMotionWrapper
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
王骏浩
ComfyUI-MimicMotionWrapper
Commits
cb5fb903
提交
cb5fb903
authored
7月 02, 2024
作者:
kijai
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
switch to relative imports
上级
e90ac6dd
隐藏空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
6 行增加
和
8 行删除
+6
-8
nodes.py
nodes.py
+6
-8
没有找到文件。
nodes.py
浏览文件 @
cb5fb903
import
os
import
os
import
torch
import
torch
import
sys
import
numpy
as
np
import
numpy
as
np
import
gc
import
gc
...
@@ -13,13 +12,12 @@ from diffusers.schedulers import EulerDiscreteScheduler
...
@@ -13,13 +12,12 @@ from diffusers.schedulers import EulerDiscreteScheduler
from
transformers
import
CLIPImageProcessor
,
CLIPVisionModelWithProjection
from
transformers
import
CLIPImageProcessor
,
CLIPVisionModelWithProjection
script_directory
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
script_directory
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
sys
.
path
.
append
(
script_directory
)
from
mimicmotion.pipelines.pipeline_mimicmotion
import
MimicMotionPipeline
from
.
mimicmotion.pipelines.pipeline_mimicmotion
import
MimicMotionPipeline
from
mimicmotion.modules.unet
import
UNetSpatioTemporalConditionModel
from
.
mimicmotion.modules.unet
import
UNetSpatioTemporalConditionModel
from
mimicmotion.modules.pose_net
import
PoseNet
from
.
mimicmotion.modules.pose_net
import
PoseNet
from
lcm_scheduler
import
AnimateLCMSVDStochasticIterativeScheduler
from
.
lcm_scheduler
import
AnimateLCMSVDStochasticIterativeScheduler
class
MimicMotionModel
(
torch
.
nn
.
Module
):
class
MimicMotionModel
(
torch
.
nn
.
Module
):
def
__init__
(
self
,
base_model_path
,
lcm
=
False
):
def
__init__
(
self
,
base_model_path
,
lcm
=
False
):
...
@@ -247,8 +245,8 @@ class MimicMotionGetPoses:
...
@@ -247,8 +245,8 @@ class MimicMotionGetPoses:
def
process
(
self
,
ref_image
,
pose_images
,
include_body
,
include_hand
,
include_face
):
def
process
(
self
,
ref_image
,
pose_images
,
include_body
,
include_hand
,
include_face
):
device
=
mm
.
get_torch_device
()
device
=
mm
.
get_torch_device
()
from
mimicmotion.dwpose.util
import
draw_pose
from
.
mimicmotion.dwpose.util
import
draw_pose
from
mimicmotion.dwpose.dwpose_detector
import
DWposeDetector
from
.
mimicmotion.dwpose.dwpose_detector
import
DWposeDetector
yolo_model
=
"yolox_l.onnx"
yolo_model
=
"yolox_l.onnx"
dw_pose_model
=
"dw-ll_ucoco_384.onnx"
dw_pose_model
=
"dw-ll_ucoco_384.onnx"
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论