mirror of
https://github.com/MarcosRodrigoT/ViT-Face-Recognition.git
synced 2025-12-30 08:02:29 +00:00
25 lines
900 B
Python
25 lines
900 B
Python
import os
|
|
|
|
|
|
base_dir = '/mnt/Data/mrt/RealWorldOccludedFaces/images'
|
|
masked_dir = f"{base_dir}/masked"
|
|
neutral_dir = f"{base_dir}/neutral"
|
|
sunglasses_dir = f"{base_dir}/sunglasses"
|
|
|
|
masked_items = os.listdir(masked_dir)
|
|
neutral_items = os.listdir(neutral_dir)
|
|
sunglasses_items = os.listdir(sunglasses_dir)
|
|
|
|
for item in masked_items:
|
|
if not os.listdir(f"{masked_dir}/{item}"):
|
|
print(f'Masked folder ->\t{item} has no images, removing directory')
|
|
os.rmdir(f'{masked_dir}/{item}')
|
|
for item in neutral_items:
|
|
if not os.listdir(f"{neutral_dir}/{item}"):
|
|
print(f'Neutral folder ->\t{item} has no images, removing directory')
|
|
os.rmdir(f'{neutral_dir}/{item}')
|
|
for item in sunglasses_items:
|
|
if not os.listdir(f"{sunglasses_dir}/{item}"):
|
|
print(f'Sunglasses folder ->\t{item} has no images, removing directory')
|
|
os.rmdir(f'{sunglasses_dir}/{item}')
|