Hugging Face's logo

briaai
/
RMBG-2.0

Image Segmentation
Transformers
PyTorch
ONNX
Safetensors
Transformers.js
remove background
background
background-removal
Pytorch
vision
legal liability
custom_code
Model card Files
xet
 Community
49

How to train on my custom dataset

#27
by ndkhanh95 - opened
Discussion
ndkhanh95

What a great work.
I would like to know more about how to train this model on my own custom dataset. Could you please give a tutorial on this? I would appreciate it.
SagiF
 BRIA AI org

Hi @ndkhanh95 ,
Sorry for the delayed response, we will create such tutorial and upload it. Timeline is probably towards the end of the month due to holidays.

SagiF
 BRIA AI org

Hi @ndkhanh95 , until we make a dedicate tutorial, please see this guide as an example on how to train our model-
https://github.com/Bria-AI/GTC25_Demo/blob/dev/gtc_demo_fine_tune_on_prem.ipynb

Let me know if you have any questions!
cmbhatt

Any updates on this?
SagiF
 BRIA AI org

Hi @cmbhatt , are you referring to the recording tutorial? We haven't record it yet. did you get the chance to look at the step by step I sent?

Sign up or log in to comment