File size: 1,722 Bytes
d972a8d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
"""
Author: Hongsong Wang
Contact: [email protected]
Date: 2025-10-21
Description: Process a color image per B, G, R channel using the lowest 3 bit-planes.
"""
import cv2
import numpy as np
import os
from tqdm import tqdm
def process_single_image(img):
# Ensure input is a numpy array with uint8
img = np.asarray(img, dtype=np.uint8)
# Validate shape
if img.ndim != 3 or img.shape[2] != 3:
raise ValueError(f"Expected color image with shape (H, W, 3), got {img.shape}")
# Extract the lowest 3 bits for each channel (bit planes 0..2)
low3 = img & 0b111 # shape (H, W, 3), values 0..7
# For each channel, check if any of the low-3 bits is non-zero -> boolean mask
mask = (low3 != 0).astype(np.uint8) # 0 or 1, shape (H, W, 3)
# Scale to 0 or 255 for a displayable binary image per channel
result = mask * 255 # dtype uint8
return result
def process_folder(input_dir, output_dir):
os.makedirs(output_dir, exist_ok=True)
exts = ('.png', '.jpg', '.jpeg', '.bmp')
image_files = [f for f in os.listdir(input_dir) if f.lower().endswith(exts)]
for fname in tqdm(image_files, desc="Processing images"):
input_path = os.path.join(input_dir, fname)
output_path = os.path.join(output_dir, fname)
img = cv2.imread(input_path)
if img is None:
print(f"Skip: {fname}")
continue
result = process_single_image(img)
cv2.imwrite(output_path, result)
if __name__ == "__main__":
input_dir = "D:/BaiduNetdiskDownload/Image/AI-generated"
output_dir = "D:/BaiduNetdiskDownload/Image/output"
process_folder(input_dir, output_dir)
|