Open zhujingsong opened 2 years ago
Thanks so much for the advice on getitem(self, index). I have been unaware of this part. And now that I know, I have edited the code.
I really appreciate your interest in my code.
Normally we will put the augmentation part in the def getitem(self, index) of class TensorData_scale(Dataset), so that for every epoch, a different kind of augmentation could be applied to [img1, img2, mask].
I found that you did the augmentation before def getitem(self, index) and saved it as a fixed numpy array. I assume possibly such a data loading way might cause out of memory problem and fixed augmentation for patch image in every epoch.
您好zhujingsong,最近看到了您编写的https://github.com/zhujingsong/Hooknet,关于数据处理有些问题想要咨询您,但是您没有开放issues,请问怎么才能联系到您呢
Normally we will put the augmentation part in the def getitem(self, index) of class TensorData_scale(Dataset), so that for every epoch, a different kind of augmentation could be applied to [img1, img2, mask].
I found that you did the augmentation before def getitem(self, index) and saved it as a fixed numpy array. I assume possibly such a data loading way might cause out of memory problem and fixed augmentation for patch image in every epoch.