search

xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.34k stars 1.43k forks source link

Is there any tutorial on how to train the model using custom dataset? #185

Open mnayef95 opened 3 years ago

mnayef95 commented 3 years ago

I was searching for a tutorial on how to train the model using a custom dataset I didn't find anything related to this topic, is there any document or tutorial that points out how to train the model?

Also is there any reference on how the dataset should look like?

5p4r70n commented 3 years ago

I'm also searching for that....

mediarl commented 3 years ago

Looking for a tutorial as well

xuebinqin commented 3 years ago

Please refer to our paper: https://arxiv.org/pdf/2005.09007.pdf Sec. 4.1 introduces the datasets. Sec. 4.4 gives the details of implementation and training.

On Tue, May 18, 2021 at 8:15 PM mediarl @.***> wrote:

Looking for a tutorial as well

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/xuebinqin/U-2-Net/issues/185#issuecomment-843315560, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADSGORNL5A7F332ZBVUUS3TTOKHCRANCNFSM42STIUJQ .

-- Xuebin Qin PhD Department of Computing Science University of Alberta, Edmonton, AB, Canada Homepage:https://webdocs.cs.ualberta.ca/~xuebin/