crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove img_w arg from transform function

Browse Source
This commit is contained in:
darkliang
2023-02-07 13:37:09 +08:00
parent bda25e8978
commit 311bc848fa
2 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
opengait/data/transform.py
+7 -8
View File
@@ -10,8 +10,8 @@ class NoOperation():
class BaseSilTransform():
def __init__(self, disvor=255.0, img_shape=None):
self.disvor = disvor
def __init__(self, divsor=255.0, img_shape=None):
self.divsor = divsor
self.img_shape = img_shape
def __call__(self, x):
@@ -19,22 +19,21 @@ class BaseSilTransform():
s = x.shape[0]
_ = [s] + [*self.img_shape]
x = x.reshape(*_)
return x / self.disvor
return x / self.divsor
class BaseSilCuttingTransform():
def __init__(self, img_w=64, disvor=255.0, cutting=None):
self.img_w = img_w
self.disvor = disvor
def __init__(self, divsor=255.0, cutting=None):
self.divsor = divsor
self.cutting = cutting
def __call__(self, x):
if self.cutting is not None:
cutting = self.cutting
else:
cutting = int(self.img_w // 64) * 10
cutting = int(x.shape[-1] // 64) * 10
x = x[..., cutting:-cutting]
return x / self.disvor
return x / self.divsor
class BaseRgbTransform():