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

Object detection for Human only #155

Open tiwari93 opened 3 years ago

tiwari93 commented 3 years ago

Hi Nathan,

Thanks for the awesome work, just one question can we modify the model or any parameter in model so that it can do detection only for human not other objects. Is there a way to classify it?

ROBYER1 commented 3 years ago

I would be interested in this also, did you find any way to do this?

xuebinqin commented 3 years ago

Hi, we just updated another model for human segmentation. Pls take a look. https://github.com/NathanUA/U-2-Net.

Thanks for your interests and support.

On Sat, Feb 6, 2021 at 1:34 PM ROBYER1 notifications@github.com wrote:

I would be interested in this also, did you find any way to do this?

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

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

ROBYER1 commented 3 years ago

@NathanUA This is fantastic work Nathan, thankyou - is there a smaller model for running on mobile devices such as Android and IOS? I assume the unetp model as that is vastly smaller

I am currently looking into converting the model to ONNX to run in Unity Game Engine to use webcams/smartphone cameras with Unity Barracuda - using this very simple example https://github.com/AlexRibard/Barracuda-U-2-NetTest

I look forward to sharing the results back here!

tiwari93 commented 3 years ago

Hi, we just updated another model for human segmentation. Pls take a look. https://github.com/NathanUA/U-2-Net. Thanks for your interests and support. On Sat, Feb 6, 2021 at 1:34 PM ROBYER1 @.***> wrote: I would be interested in this also, did you find any way to do this? — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#155 (comment)>, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADSGORI6IR6JLB753WSQVHTS5WRVBANCNFSM4WRFC5YA . -- Xuebin Qin PhD Department of Computing Science University of Alberta, Edmonton, AB, Canada Homepage:https://webdocs.cs.ualberta.ca/~xuebin/

Thanks for your help. Will try this and update you for the same

Siq1982 commented 3 years ago

Me too

@NathanUA This is fantastic work Nathan, thankyou - is there a smaller model for running on mobile devices such as Android and IOS? I assume the unetp model as that is vastly smaller

I am currently looking into converting the model to ONNX to run in Unity Game Engine to use webcams/smartphone cameras with Unity Barracuda - using this very simple example https://github.com/AlexRibard/Barracuda-U-2-NetTest

I look forward to sharing the results back here!