我现在尝试用BirefNet.safetensor进行扣像处理，发现有很多的weights，不知道weights应该怎么选取。

ZhengPeng7 / BiRefNet

[CAAI AIR'24] Bilateral Reference for High-Resolution Dichotomous Image Segmentation

https://www.birefnet.top

MIT License

1.42k stars 105 forks source link

Closed ZzMLvzZ-792998470 closed 2 days ago

ZhengPeng7 commented 2 days ago

如果没有特定场景需求, 就顾名思义, 用general.

ZzMLvzZ-792998470 commented 2 days ago

如果没有特定场景需求, 就顾名思义, 用general. 不好意思哈，没表达清楚，是这个weights的选取我不太清楚

ZhengPeng7 commented 2 days ago

这个weights是backbone的weights, 如果你不需要训练的话, 那其实可以不用管这个的. 在创建时设下参数不调用就可以: BiRefNet(bb_pretrained=False).

ZzMLvzZ-792998470 commented 2 days ago

这个weights是backbone的weights, 如果你不需要训练的话, 那其实可以不用管这个的. 在创建时设下参数不调用就可以: BiRefNet(bb_pretrained=False).

原来如此感谢大佬！