Train my dataset [Value error] wrong input array

[milumilule2]

Train my dataset [Value error] wrong input array

[dokutagero]

I am still learning with the code of this implementation so my advice could not be right. As far as I know, if you are using the SegDataGenerator, the default way in this repo, for loading your images and labels, note that it is specified in the docstring of that class that the labels must be one channel with pixels specifying the class they belong to. This is in case that reshape error has to do with the label format.

[milumilule2]

@dokutagero Thank you so much for your reply! In fact, I want to use this implementation to train my saliency detection dataset. In detail, the input images are RGB images(channel=3), but the output are grey images(channel=1). So the results are not classification for the objects in the image. The results are recognizing the saliency object in the image. After training, the saliency object should be white and the background should be black in the output map. Do you think this implementation can solve my task? Can I finish my task after modifying some parameters? Please give me some advice. Thank you again!

[JingangLang]

Hi,friends. @milumilule2 .I'm also trying to find suitful way to train my saliency detection dataset,but all the values in the final prediction are 255 which means the final map is a white picture.I guess that the loss function maybe the reason.However,I have no ideas to solve the problem.Can you give me some advice.Thank you!Happy to communicate with you.Best wishes!

[chinmayshah99]

Hey @JingangLang check out #2

[watchmexiang]

Hello, I also have this problem, I want to train my data set, but the picture of my data set is jpg, the label is png, can I use this data set for training, if I can, I need the program What changes have been made?thanks