MCPcopy Index your code
hub / github.com/tensorpack/tensorpack / _aug_mapper

Method _aug_mapper

tensorpack/dataflow/image.py:157–167  ·  view source on GitHub ↗
(self, dp)

Source from the content-addressed store, hash-verified

155 self.augs.reset_state()
156
157 def _aug_mapper(self, dp):
158 with self._exception_handler.catch():
159 img, coords = dp[self._img_index], dp[self._coords_index]
160 check_dtype(img)
161 validate_coords(coords)
162 if self._copy:
163 img, coords = copy_mod.deepcopy((img, coords))
164 tfms = self.augs.get_transform(img)
165 dp[self._img_index] = tfms.apply_image(img)
166 dp[self._coords_index] = tfms.apply_coords(coords)
167 return dp
168
169
170class AugmentImageComponents(MapData):

Callers

nothing calls this directly

Calls 6

check_dtypeFunction · 0.85
validate_coordsFunction · 0.85
catchMethod · 0.80
get_transformMethod · 0.45
apply_imageMethod · 0.45
apply_coordsMethod · 0.45

Tested by

no test coverage detected