Skip to content

Commit

Permalink
Add fill value to cutout (#284)
Browse files Browse the repository at this point in the history
* added fill value to Cutout constructory

* fixed except clause

* flake8 fixes

* Delete test.py

dummy file

* Update transforms_interface.py

remove unnecessary if statements

* added fill value to CoarseDropouty

* fixed update params

* switched fill_value to optional in apply()

* fixing test fails

* fixed signatures

* flake8 fixes

* more flake8 fixes

* fixed mistake in params dictionary
  • Loading branch information
bfialkoff authored and ternaus committed Jul 2, 2019
1 parent 611d324 commit d85bab5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion albumentations/core/transforms_interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ def update_params(self, params, **kwargs):
if hasattr(self, 'interpolation'):
params['interpolation'] = self.interpolation
if hasattr(self, 'fill_value'):
params['interpolation'] = self.fill_value
params['fill_value'] = self.fill_value
params.update(
{'cols': kwargs['image'].shape[1],
'rows': kwargs['image'].shape[0]})
Expand Down

0 comments on commit d85bab5

Please sign in to comment.