how to edit in train.py to load in my own images and labels?

[zhou-rui1]

I got train images: 0 train labels: 0 I set my folders like this: ––DUTS-TR ——im_aug ———0.jpg ———1.png ——gt_aug ———0.png ———1.png is there something wrong?

ValueError: num_samples should be a positive integer value, but got num_samples=0

what should I edit, any guidance will be helpful

`data_dir = os.path.join(os.getcwd(), 'train_data' + os.sep)

  | tra_image_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'im_aug' + os.sep)   | tra_label_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'gt_aug' + os.sep)`

[xuebinqin]

you can print tra_image_dir to see if you put the DUTS under the correct directory (train_data).

On Sat, Nov 21, 2020 at 6:36 AM zhou-rui1 notifications@github.com wrote:

I got train images: 0 train labels: 0 I set my folders like this: ––DUTS-TR ——im_aug ———0.jpg ———1.png ——gt_aug ———0.png ———1.png is there something wrong? ValueError: num_samples should be a positive integer value, but got num_samples=0 what should I edit, any guidance will be helpful data_dir = os.path.join(os.getcwd(), 'train_data' + os.sep)

| tra_image_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'im_aug' + os.sep) | tra_label_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'gt_aug' + os.sep)

— 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/96, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADSGORJBYBXAI4TL47EU2KDSQ664NANCNFSM4T5ZJZTQ .

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

[Chints7323]

I got train images: 0 train labels: 0 I set my folders like this: ––DUTS-TR ——im_aug ———0.jpg ———1.png ——gt_aug ———0.png ———1.png is there something wrong?

ValueError: num_samples should be a positive integer value, but got num_samples=0

what should I edit, any guidance will be helpful

`data_dir = os.path.join(os.getcwd(), 'train_data' + os.sep)

  | tra_image_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'im_aug' + os.sep)   | tra_label_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'gt_aug' + os.sep)`

im_aug should have all images in '.jpg' format and gt_aug should have all in '.png'

[xuebinqin]

you can set "tra_image_dir=/your path/" directly instead of using OS.path.join

On Thursday, March 17, 2022, Chints7323 @.***> wrote:

I got train images: 0 train labels: 0 I set my folders like this: ––DUTS-TR ——im_aug ———0.jpg ———1.png ——gt_aug ———0.png ———1.png is there something wrong?

ValueError: num_samples should be a positive integer value, but got num_samples=0

what should I edit, any guidance will be helpful `data_dir = os.path.join(os.getcwd(), 'train_data' + os.sep)

| tra_image_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'im_aug' + os.sep) | tra_label_dir = os.path.join('DUTS', 'DUTS-TR', 'DUTS-TR', 'gt_aug' + os.sep)`

im_aug should have all images in '.jpg' format and gt_aug should have all in '.png'

— Reply to this email directly, view it on GitHub https://github.com/xuebinqin/U-2-Net/issues/96#issuecomment-1071068894, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADSGORKK62L6QJGT4V7MYSDVANOMBANCNFSM4T5ZJZTQ . Triage notifications on the go with GitHub Mobile for iOS https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675 or Android https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub.

You are receiving this because you commented.Message ID: @.***>

-- Xuebin Qin PhD Department of Computing Science University of Alberta, Edmonton, AB, Canada Homepage: https://xuebinqin.github.io/