-
Hi, I'd like to try this in my u2net imagenet model.
Can i simply drop it in and replace Adam/SGD?
I noticed that you are calling net.zero_grad() in your training loop, however in my script its…
-
你好 想加入mae(绝对误差指标)但是输出图片通道数为num_class,而我单纯就是二分类怎么解决呢
-
I am a student of Tsinghua University. I want to ask whether u2net only supports single category segmentation. For example, for the picture of people riding on bicycles, can people and bicycles be sep…
-
最近刚刚开始学习object detection,谢谢大佬的开源!之前直接用u2net在自己的dataset上跑结果很好~可是如果加了完全是背景的图片结果就很奇怪, 所以自己试着想改成classification的问题,我把sigmoid换成了softmax, loss换成了cross entropy lossoutput channel也是变成了2,感觉这样是不是就把background 和 o…
-
Thanks for your great work! I'm trying to train u2net for human segmentation, but the image size for further inference is 1920*2560. The given pretrained model does not perform well. Would you please …
-
hi qin, I open this issue to ask why not use ssim and iou loss which used in BASNet?It seems they are effective in BASBet, shown in Ablation Study part. Could you please tell me why you abandon them i…
-
Hi~Mr Qin,I am coming again....
I retrain u2net on my datasets(9000 pics),the loss value oscillate around 0.2,do you think should I keep training??(I just train for 1.5 day),I am a little worried tha…
-
@SwatiModi Thanks for such a nice work. I have a question related to the training model . I am working same sort of the problem , so I am using u2net segmentation Model for my solution . But it is rea…
-
Hi!
I would like to share one [small tool](https://github.com/peko/nn-lineart) that you may found useful. Right now I use it to compare differences between u2net and artline, it allows to quickly s…
-
I want to use the API on webui server with the payload as :
payload = {
"image": encode_pil_to_base64(image),
"model": "u2net",
"return_mask": False,
"alpha_matting": False,
…