(self, config,trans_type)
| 98 | |
| 99 | class alignCollate(object): |
| 100 | def __init__(self, config,trans_type): |
| 101 | self.imgH = config['train']['imgH'] |
| 102 | self.imgW = config['train']['imgW'] |
| 103 | self.use_tia = config['train']['use_tia'] |
| 104 | self.aug_prob = config['train']['aug_prob'] |
| 105 | self.label_transform = strLabelConverter(config['train']['alphabet']) |
| 106 | self.trans_type = trans_type |
| 107 | self.isGray = config['train']['isGray'] |
| 108 | self.ConAug = config['train']['ConAug'] |
| 109 | |
| 110 | def __call__(self, batch): |
| 111 | images, labels = zip(*batch) |
nothing calls this directly
no outgoing calls
no test coverage detected