# import the modules
import os
from os import listdir

# get the path/directory
folder_dir = r"E:\Kamal\debugging\attentionGAN\tools_tf2\dataset\test.txt"
for images in os.listdir(folder_dir):

	# check if the image ends with png
	if (images.endswith("rain.png")):
		print(images)

# import the modules
import os
from os import listdir

# get the path/directory
folder_dir = r"E:\Kamal\debugging\attentionGAN\tools_tf2\dataset\test.txt"
for images in os.listdir(folder_dir):

	# check if the image ends with png
	if (images.endswith("rain.png")):
		print(images)

# import the modules # get the path/directory # check if the image ends with png