import os

if __name__ == "__main__":
    # 读取第一个txt文件的第一列文件名到一个集合中
    file_set1 = set()
    with open("/media/DataDrive/yiling/annotation/XD_test_annotation.txt", "r") as file1:
        for line in file1:
            filename = line.strip().split()[0]  # 假设每行的数据以空格分隔
            filename = os.path.basename(filename)
            file_set1.add(filename)

    # 读取第二个txt文件的第一列文件名到一个集合中
    file_set2 = set()
    with open("/media/DataDrive/yiling/annotation/XD_train_annotation.txt", "r") as file2:
        for line in file2:
            filename = line.strip().split()[0]  # 假设每行的数据以空格分隔
            filename = os.path.basename(filename)
            file_set2.add(filename)

    # 检查两个集合的交集
    common_files = file_set1.intersection(file_set2)
    cnt = len(common_files)
    # 如果交集不为空，则表示两个文件中有相同的文件名
    if common_files:
        print("两个文件中有相同的文件: ", cnt)
    else:
        print("两个文件中没有相同的文件名.")