jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[SubImage]

Browse Source 
- handle different image shapes
This commit is contained in:
jens
2026-01-11 11:21:42 +01:00
parent 3f4cdc277f
commit 92afc16137
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/image.py
+7 -3
View File
@@ -13,8 +13,6 @@ class SubImage:
self.sy = sy
self.px = px
self.py = py
self.x = 0
self.y = 0
def check(self, image: Mat):
nx = (image.shape[0] - self.sx)
@@ -28,7 +26,13 @@ class SubImage:
ny = int(ny / self.py) + 1
return nx, ny
def __call__(self, images: Mat):
def __call__(self, images: Mat, n_ch: int = 3):
if len(images.shape) == 3:
return self.process_image(np.reshape(images, (images.shape[0], n_ch) + images.shape[1:3]))
if len(images.shape) == 2:
return self.process_image(np.reshape(images, (1,n_ch,) + images.shape)), images.shape
return self.process_image(images)
def process_image(self, image: Mat):