patchy/keys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove falsely keyed out noise as well

Browse Source
This commit is contained in:
Ian Gulliver
2024-08-18 06:07:51 -07:00
parent 1ef083dbaf
commit 0aaee5eaea
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
chromakey.py
View File

@@ -62,11 +62,13 @@ def _islands(mask):
def _filter_islands(mask, ids, min_object_pixels): def _filter_islands(mask, ids, min_object_pixels):
values, counts = np.unique(ids.flatten(), return_counts=True) values, counts = np.unique(ids.flatten(), return_counts=True)
ix = np.where(counts < min_object_pixels) values_ix = np.where(counts < min_object_pixels)
mask[np.isin(ids, values[ix])] = 0 mask_ix = np.isin(ids, values[values_ix])
mask[mask_ix] = np.where(mask[mask_ix] == 255, 0, 255)
def _flood_fill(ids, mask, i, j, val): def _flood_fill(ids, mask, i, j, val):
base_val = mask[i, j]
queue = [(i, j)] queue = [(i, j)]
while queue: while queue:
@@ -75,7 +77,7 @@ def _flood_fill(ids, mask, i, j, val):
if i < 0 or i >= mask.shape[0] or j < 0 or j >= mask.shape[1]: if i < 0 or i >= mask.shape[0] or j < 0 or j >= mask.shape[1]:
continue continue
if mask[i, j] == 0 or ids[i, j] != 0: if mask[i, j] != base_val or ids[i, j] != 0:
continue continue
ids[i, j] = val ids[i, j] = val
@@ -107,4 +109,4 @@ filename = os.path.join(path, '41244/back.orf')
with Image(filename=filename) as img: with Image(filename=filename) as img:
img.crop(left=0, top=0, width=4000, height=3000) img.crop(left=0, top=0, width=4000, height=3000)
chroma_key(img) chroma_key(img)
img.save(filename='chromakey.png') img.save(filename='output.png')