Skip to content
Snippets Groups Projects

WIP: [Transform class] add ResizeCrop for drhagis

Closed Driss KHALIL requested to merge DrhagisUpdates into master
@@ -291,10 +291,10 @@ class ResizeCrop:
Parameters
-------------
*args: 3 sized array
* args[0] : Contains the image
* args[1] : Contains the label
* args[2] : Contains the mask
args: 3-sized-array
args[0] : Contains the image
args[1] : Contains the label
args[2] : Contains the mask
"""
@@ -334,7 +334,9 @@ class ResizeCrop:
new_label = expand2square(new_label, 0)
new_mask = expand2square(new_mask, 0)
return new_img,new_label,new_mask
args = (new_img, new_label, new_mask)
return args
Loading