search

dbpprt / u-2-net-portrait

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
116 stars 14 forks source link

about train my class segmentation #8

Open xiongzhu666 opened 3 years ago

xiongzhu666 commented 3 years ago

hi，thanks for your great working，I want to train my single class segmentation based on your project，I have some doubt about training：

  1. the dataset format：the data directory should include image and mask，the mask is binary single channel image，right？？
  2. could I run train.py directly？？what else should I consider and modify？？ thanks best regard
xuebinqin commented 3 years ago
  1. image with RGB channels [0,255], Mask with one channel (0-background, 255-foreground).
  2. yes, you can train with 'train.py' directly. just modify the training data path should work. If you want, you can add the validation process in the training code.

On Mon, Jun 7, 2021 at 6:34 AM xiongzhu666 @.***> wrote:

hi，thanks for your great working，I want to train my single class segmentation based on your project，I have some doubt about training：

  1. the dataset format：the data directory should include image and mask，the mask is binary single channel image，right？？
  2. could I run train.py directly？？what else should I consider and modify？？ thanks best regard

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/dennisbappert/u-2-net-portrait/issues/8, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADSGORPD2PILRETGWXD4E43TRQV4PANCNFSM46GV5KTQ .

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

Testhjf commented 2 years ago

@xuebinqin @dennisbappert hello sir，when i train ，it propers bellow problems： hydra.errors.InstantiationException: Top level config has to be OmegaConf DictConfig, plain dict, or a Structured Config class or instance。 how can i deal with it ？ thanks