91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
# This is a sample Python script.
|
||
|
||
# Press ⌃R to execute it or replace it with your code.
|
||
# Press Double ⇧ to search everywhere for classes, files, tool windows, actions, and settings.
|
||
import numpy as np
|
||
from scipy.fft import fft2, fftshift
|
||
import cv2
|
||
import matplotlib.pyplot as plt
|
||
|
||
def print_hi(name):
|
||
# Use a breakpoint in the code line below to debug your script.
|
||
print(f'Hi, {name}') # Press ⌘F8 to toggle the breakpoint.
|
||
|
||
|
||
def encode_watermark_to_binary(watermark_sequence):
|
||
binary_sequence = []
|
||
for number in watermark_sequence:
|
||
# 初始化一个100位全为'0'的二进制字符串
|
||
binary_100_bit = ['0'] * 100
|
||
# 在数字对应的位置设置'1'
|
||
if 0 <= number < 100:
|
||
binary_100_bit[number-1] = '1'
|
||
# 连接这些位，形成一个100位的二进制字符串
|
||
binary_sequence.append(''.join(binary_100_bit))
|
||
return binary_sequence
|
||
|
||
def process_image_b_component(image_path):
|
||
"""
|
||
处理图像的B通道，执行DFT并移动频谱。
|
||
|
||
:param image_path: 输入图像的路径。
|
||
:return: 移位后DFT的幅度矩阵。
|
||
"""
|
||
# 使用合适的库加载图像（例如，PIL, OpenCV）
|
||
# 这里我们会加载图像并提取B通道
|
||
# 作为演示，我们将创建一个假的256x256 B通道矩阵
|
||
|
||
#根据image_path加载图像
|
||
image = cv2.imread(image_path)
|
||
# 检查图像是否成功加载
|
||
if image is None:
|
||
raise ValueError("无法加载图像，请检查路径是否正确。")
|
||
|
||
# 确保图像的大小是256x256
|
||
image_resized = cv2.resize(image, (256, 256))
|
||
|
||
# 提取B通道
|
||
b_channel = image_resized[:, :, 0]
|
||
|
||
# 对B通道矩阵执行DFT
|
||
dft_result = fft2(b_channel)
|
||
|
||
# 将零频率分量移动到频谱的中心
|
||
dft_shifted = fftshift(dft_result)
|
||
|
||
# 获取移位后DFT的幅度
|
||
magnitude = np.abs(dft_shifted)
|
||
|
||
return magnitude
|
||
|
||
def decode_binary_to_watermark(binary_sequence):
|
||
"""
|
||
Decode the binary sequence back into the original array of numbers.
|
||
|
||
:param binary_sequence: A list of 100-bit strings representing the binary codes.
|
||
:return: The original array of numbers.
|
||
"""
|
||
original_numbers = []
|
||
for binary_code in binary_sequence:
|
||
if '1' in binary_code:
|
||
# Find the index of '1' and convert it back to the original number
|
||
position = binary_code.index('1') + 1 # Add 1 because the index starts at 0
|
||
original_number = position
|
||
original_numbers.append(original_number)
|
||
return original_numbers
|
||
|
||
# Press the green button in the gutter to run the script.
|
||
if __name__ == '__main__':
|
||
# print_hi('PyCharm')
|
||
watermark_sequence = [12, 34, 56, 44, 90, 55, 34, 56]
|
||
result = encode_watermark_to_binary(watermark_sequence)
|
||
print(result)
|
||
# # 打印编码后的二进制序列
|
||
# for binary in result:
|
||
# print(binary)
|
||
|
||
watermark_sequence = decode_binary_to_watermark(result)
|
||
print(watermark_sequence)
|
||
|
||
# See PyCharm help at https://www.jetbrains.com/help/pycharm/
|