Custom datasets

[mouxinyue1]

Hello, I am using a custom data set, because my data set has a transparency, so it does not match, I would like to ask how to solve this problem?

[gudovskiy]

@mouxinyue1 sorry, cannot help with this description

[sangbin-kim]

@gudovskiy Hello, related question, I would like to train my custom dataset. It would be very helpful if you could explain the dataset structure for both train and test sets.

[gudovskiy]

@sangbin-kim please refer to this description and our dataloader code

[sangbin-kim]

@gudovskiy Thank you sir. is there any way to train without ground truth dataset for this model??

[gudovskiy]

@sangbin-kim you don't need ground truth for train data, only for test data to check performance

[EudicL]

Hello, I am using a custom data set, because my data set has a transparency, so it does not match, I would like to ask how to solve this problem?

I also encountered this problem, how did you solve it?

[gudovskiy]

@EudicL what's the file and line number where this happens?

[EudicL]

Thanks, I have solved the problem. Because the images in my datasets are gray, I modify the [loader.py]
'''
x = Image.open(x).convert('RGB')
'''

@EudicL what's the file and line number where this happens?

[gudovskiy]

@EudicL great, may be I should add the check for such case