Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
5f12e7ef
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
大约 1 年 前同步成功
通知
1784
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
提交
5f12e7ef
编写于
10月 07, 2022
作者:
G
Greendayle
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
linux test
上级
fa2ea648
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
3 addition
and
2 deletion
+3
-2
modules/deepbooru.py
modules/deepbooru.py
+3
-2
未找到文件。
modules/deepbooru.py
浏览文件 @
5f12e7ef
import
os.path
import
os.path
from
concurrent.futures
import
ProcessPoolExecutor
from
concurrent.futures
import
ProcessPoolExecutor
from
multiprocessing
import
get_context
def
_load_tf_and_return_tags
(
pil_image
,
threshold
):
def
_load_tf_and_return_tags
(
pil_image
,
threshold
):
...
@@ -66,7 +66,8 @@ def subprocess_init_no_cuda():
...
@@ -66,7 +66,8 @@ def subprocess_init_no_cuda():
def
get_deepbooru_tags
(
pil_image
,
threshold
=
0.5
):
def
get_deepbooru_tags
(
pil_image
,
threshold
=
0.5
):
with
ProcessPoolExecutor
(
initializer
=
subprocess_init_no_cuda
)
as
executor
:
context
=
get_context
(
'spawn'
)
with
ProcessPoolExecutor
(
initializer
=
subprocess_init_no_cuda
,
mp_context
=
context
)
as
executor
:
f
=
executor
.
submit
(
_load_tf_and_return_tags
,
pil_image
,
threshold
,
)
f
=
executor
.
submit
(
_load_tf_and_return_tags
,
pil_image
,
threshold
,
)
ret
=
f
.
result
()
# will rethrow any exceptions
ret
=
f
.
result
()
# will rethrow any exceptions
return
ret
return
ret
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录