xseg apply失败

wellingyep

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

提示信息：ValueError: cannot reshape array of size 294912 into shape (3,3,128,128)
按网上说似乎是图片大小的问题，那怎么去改呢，提取图片的时候确认和xseg一样是wf没问题，还是说应该提取成jpg而不能是png呢？
xseg试了GenericXSeg和论坛里的https://dfldata.xyz/forum.php?mo ... &highlight=xseg都是同样的问题
================== Model Summary ==================
==                                               ==
==        Model name: XSeg                       ==
==                                               ==
== Current iteration: 323192                     ==
==                                               ==
==---------------- Model Options ----------------==
==                                               ==
==         face_type: wf                         ==
==        batch_size: 16                         ==
==                                               ==
==----------------- Running On ------------------==
==                                               ==
==      Device index: 0                          ==
==              Name: NVIDIA GeForce RTX 2080 Ti ==
==              VRAM: 9.32GB                     ==
==                                               ==
===================================================
上面是GenericXSeg，下面是论坛里的
============= Model Summary ===============                                     ====        Model name: XSeg             ====                                     ==== Current iteration: 676694           ====                                     ====----------- Model Options -----------====                                     ====         face_type: wf               ====        batch_size: 8                ====                                     ====------------ Running On -------------====                                     ====      Device index: 0                ====              Name: GeForce GTX 1650 ====              VRAM: 2.86GB           ====                                     ===========================================
我的模型也是wf

FINITEIN

路过水一贴

lalakia2012

知而不言，乃君子所爲！

wellingyep

lalakia2012 发表于 2021-10-29 16:32
知而不言，乃君子所爲！

wellingyep

已解决，是colab版本和遮罩版本不对应的问题，在佬们眼里应该是低级错误了吧
用最新版就好了，其实遮罩的原贴主也说了是最新版，就是要翻十几页才能看到这句话，所以一开始就错过了

luofeiyu

路过学习

kayagood

看不来参数

sdsi

apply是什么意思啊？

wellingyep

sdsi 发表于 2021-10-30 01:30
apply是什么意思啊？

就是把xseg 加到dst里去，这样xseg就可以在训练的时候生效
colab 里面有Apply or remove XSeg mask to the faces

liu3307307

咋解决啊 我也遇到这问题了？软件版本我082的 脸WF 512的