You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
2.5 KiB
Python
55 lines
2.5 KiB
Python
2 weeks ago
|
import os
|
||
|
|
|
||
|
|
def split_combined_image(combined_image_path, output_folder):
|
||
|
|
# Read the combined image in binary mode
|
||
|
|
with open(combined_image_path, 'rb') as combined_image_file:
|
||
|
|
combined_data = combined_image_file.read()
|
||
|
|
|
||
|
|
# Find the JPEG end marker (FFD9 in hex) in the combined image data
|
||
|
|
jpeg_end_marker = b'\xFF\xD9'
|
||
|
|
end_marker_index = combined_data.find(jpeg_end_marker)
|
||
|
|
|
||
|
|
if end_marker_index == -1 or end_marker_index == len(combined_data) - 2:
|
||
|
|
# No second image found if the end marker is not found or it is at the end of file
|
||
|
|
print(f"Image '{combined_image_path}' cannot be split; no appended data found.")
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Split the combined data at the end marker
|
||
|
|
first_image_data = combined_data[:end_marker_index + 2] # Include the marker
|
||
|
|
second_image_data = combined_data[end_marker_index + 2:] # Everything after the marker
|
||
|
|
|
||
|
|
# Save the first part as the first image
|
||
|
|
first_image_name = os.path.basename(combined_image_path).replace('.jpg', '_part1.jpg')
|
||
|
|
output_first_image_path = os.path.join(output_folder, first_image_name)
|
||
|
|
with open(output_first_image_path, 'wb') as first_image_file:
|
||
|
|
first_image_file.write(first_image_data)
|
||
|
|
print(f"First image data saved as '{output_first_image_path}'.")
|
||
|
|
|
||
|
|
# Save the second part as the second image
|
||
|
|
second_image_name = os.path.basename(combined_image_path).replace('.jpg', '_part2.jpg')
|
||
|
|
output_second_image_path = os.path.join(output_folder, second_image_name)
|
||
|
|
with open(output_second_image_path, 'wb') as second_image_file:
|
||
|
|
second_image_file.write(second_image_data)
|
||
|
|
print(f"Second image data saved as '{output_second_image_path}'.")
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
def split_images_in_directory(input_folder, output_folder):
|
||
|
|
# Ensure the output folder exists
|
||
|
|
if not os.path.exists(output_folder):
|
||
|
|
os.makedirs(output_folder)
|
||
|
|
|
||
|
|
# Iterate over each image in the input folder
|
||
|
|
for filename in os.listdir(input_folder):
|
||
|
|
if filename.lower().endswith(('.jpg', '.jpeg', '.png')):
|
||
|
|
image_path = os.path.join(input_folder, filename)
|
||
|
|
print(f"Attempting to split '{image_path}'...")
|
||
|
|
|
||
|
|
# Attempt to split the image
|
||
|
|
if split_combined_image(image_path, output_folder):
|
||
|
|
print(f"Successfully split '{image_path}'. Terminating script.")
|
||
|
|
#return # Exit the function after a successful split
|
||
|
|
|
||
|
|
# Usage
|
||
|
|
split_images_in_directory('images', 'output_split_images')
|