Comments (5)
I don't suggest to use CRF as the post-processing, because it often introduces many salt-pepper noises along the boundaries. If you want sharp and clear boundaries, you can try our previous BASNet (https://github.com/NathanUA/BASNet) or use the cascaded PSP (https://github.com/hkchengrex/CascadePSP) as the post-process of the U^2-Net results.
from u-2-net.
@tiwari93 please keep updated if you manage to fine-tune the model.
I've tried:
- fine-tuning with the custom dataset (300 images), but the results became poor after first iterations and haven't improved
- fine-tuning on the original dataset + our custom. The results still became worse
Also, I tried different configurations of layer freezing, with different parameters of Adam optimizer.
But couldn't produce good results
from u-2-net.
Thank u for your response and suggestion.
I will try these and share you the results.
from u-2-net.
Hi,
I just wanted to share few update.
I tried cascade psp for post processing of u2net result and results are really good but it requires GPU and also it take too much time. So here we are losing the benefit of fast processing of u2net.
I just have one thought, Is it possible to fine tune u2net pre trained model on our custom dataset.
Please suggest and thank you again for ur help
from u-2-net.
It might be easier to train from scratch on your dataset than fine-tuning
from u-2-net.
Related Issues (20)
- [Question] Trained model for proprietary use HOT 1
- Ai
- RuntimeError: unexpected EOF, The file might be corrupted
- 大佬,可以提供一下验证模型相关指标的代码吗? HOT 1
- RescaleT:Why not prioritize maintaining the aspect ratio
- Hi, when I run this code, I get strange errors in other detection tasks.The following is the warning where the error occurs HOT 1
- Continue training, train my own model with U-2-Net, in between due to some reasons the training was interrupted or I want to strengthen an existing model, what should I do? Can you provide a 'Continue-training.py'? HOT 4
- Inference speed HOT 1
- 模型训练时间过长
- 对u2netp模型进行qat量化
- How can I input video or webcam in the test.py script?
- 能否麻烦将整个项目打个包(包含运行文件和预训练模型),集成一个 .bat 文件,点击运行即可使用? HOT 2
- 请问该模型只能输出二分类结果么?可以输出为多分类么?比如三分类 HOT 1
- Is it support person segmentation now? HOT 2
- 如何添加评价指标
- ImportError: cannot import name 'U2NET' from 'model' HOT 2
- Can' Access Human Segmentation Model Weights HOT 1
- Usage on cross platform mobile devices using tenser flow or pytourch ( Special on Flutter )
- 请问大家,我需要前景而非mask,该怎样修改代码呢,感谢
- /alueror: At least one stride in the given numpy aray is mepative, ad tensors with nepative strides are not curently suported. (You can probably work around this by making a copy of your array with array.copy( ).)
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from u-2-net.