about train dataset

[luoww1992]

in datasetLodaer.py:
train_dataset = DataGenerator(
data = ImageFileTrain(
alpha_dir='path/to/alpha',
fg_dir='path/to/fg',
bg_dir='path/to/bg',
root='path/to/Adobe_Image_Matting'
),
phase = 'train'
)

i know the alpha_dir is mask path,
Q1: but the fg_dir is RGB image only fg object ? Or is RGBA image the all image with fg and bg ?
Q2：the bg_dir is RGB image only bg without fg object mask? Or likes bg in AIM with fg mask?

------ can you share me a demo dataset fold with some pictures ? thank you !

[JingfengYao]

Images in fg_dir and bg_dir are both RGB images. As for demos, maybe you can see here.

[luoww1992]

@JingfengYao
Q1, so train datase likes this:

the fg_image's background must be bg_image ?
Q2, And what i mean about demo dataset is :
can you share me three images in fg_dir,bg_dir andalpha_dir in your dataset ?

[yuqilol]

Hello, has this problem been solved? Is it your above format

[JingfengYao]

@luoww1992 @yuqilol
A1: No, In our setting, there is no requirement for the foreground and background to be associated.
A2: I can't share the demos. Because the dataset is semi-public (owned by DIM). I have no right to share them personally. You should get the dataset following their instructions.
BTW, for a simple way to get familiar with matting datasets, you may check GFM. (they are similar in format to DIM data, but we didn't use them in ViTMatte training)