65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
|
from PIL import Image
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def check_image_targets(folder_path):
|
||
|
|
"""
|
||
|
|
Check if targets (labels) of images in a folder are between 0 and 1.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
- folder_path (str): Path to the folder containing images.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
- list: List of filenames with targets outside the range [0, 1].
|
||
|
|
"""
|
||
|
|
filenames = os.listdir(folder_path)
|
||
|
|
invalid_files = []
|
||
|
|
|
||
|
|
for filename in filenames:
|
||
|
|
file_path = os.path.join(folder_path, filename)
|
||
|
|
|
||
|
|
# Load image using PIL
|
||
|
|
try:
|
||
|
|
img = Image.open(file_path)
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error loading {filename}: {str(e)}")
|
||
|
|
continue
|
||
|
|
|
||
|
|
# Assuming you have a way to extract the target value from the image or filename
|
||
|
|
target = extract_target(filename) # Replace with your own function to extract target
|
||
|
|
|
||
|
|
# Check if target is outside the range [0, 1]
|
||
|
|
if not (0 <= target <= 1):
|
||
|
|
invalid_files.append(filename)
|
||
|
|
print(f"Invalid target {target} in {filename}")
|
||
|
|
|
||
|
|
return invalid_files
|
||
|
|
|
||
|
|
|
||
|
|
def extract_target(filename):
|
||
|
|
"""
|
||
|
|
Example function to extract target from filename.
|
||
|
|
Modify this according to your actual filenames and target extraction method.
|
||
|
|
"""
|
||
|
|
# Example: Extract target assuming filename format "image_target.ext"
|
||
|
|
parts = filename.split("_")
|
||
|
|
target_str = parts[-1].split(".")[0] # Remove extension
|
||
|
|
try:
|
||
|
|
target = float(target_str)
|
||
|
|
except ValueError:
|
||
|
|
target = -1 # Invalid target value
|
||
|
|
|
||
|
|
return target
|
||
|
|
|
||
|
|
|
||
|
|
# Example usage:
|
||
|
|
folder_path = r"data/LoveDA/Test/pv/masks_png_convert"
|
||
|
|
invalid_files = check_image_targets(folder_path)
|
||
|
|
|
||
|
|
if invalid_files:
|
||
|
|
print(f"Found {len(invalid_files)} files with invalid targets:")
|
||
|
|
for file in invalid_files:
|
||
|
|
print(file)
|
||
|
|
else:
|
||
|
|
print("All targets are within the valid range [0, 1].")
|