Open pissw2016 opened 5 years ago
After debug with the full class SegmentationGenerator(Sequence):
I realize that with the official VOC2012 data set. You have to not only make your own self.image_path_list
and self.label_path_list
. You have to use label = np.array(Image.open(label_path))
and I believe that the class 21 refers to unlabeled class. And there is DA(data augmentation ) the data weight will work as long as your label load in the range of (0,21) . with opencv.imread
you will get range (0,220).
nice job!!! same issue
Hi, I am new to python and keras, so your code really help me a lot. When I am running segmentation.ipynb I got a error in data generation when trainning , so I did some debug. The error happend during the middel of training some times at step76 some times step 99 like it is some picture error, so I stop shuffle. It was still random. error:
Then I comment out :
and I notice that the
self.SW
is just in[0,1]
and not a weight consider the pixels amount relative also the model only got one input layer.sample_dict = {'pred_mask': self.SW}
I change 'return self.X, self.Y, sample_dict' toreturn self.X, self.Y
and it works. So it seems that these code is not used, right?
Data augmentation seems not working cause the step amount in an epoch equals to image amount. How can I make it working? Or it is already working and I miss it?