import os
import shutil

def check_masked_folder(base_path):
    for folder_name in os.listdir(base_path):
        folder_path = os.path.join(base_path, folder_name)
        if os.path.isdir(folder_path):
            masked_folder_path = os.path.join(folder_path, 'masked')
            if os.path.isdir(masked_folder_path):
                if '0015.png' not in os.listdir(masked_folder_path):
                    print(f"'0015.png' not found in: {masked_folder_path}")
                    shutil.rmtree(folder_path)
            else:
                print(f"'masked' folder not found in: {folder_path}")

base_path = '/data_160TB/2021/yangkun/gha-gaussian/datasets/maxed_data/train'
check_masked_folder(base_path)