#!/usr/bin/env python 
# -*- coding: utf-8 -*-
"""
@project: 
@File    : otherperations
@Author  : qiqq
@create_time    : 2023/10/8 11:03
"""

import os

'''
根据bat（最原始版本） 和非bat（新版本）之间的差集 删除  相应的原图和标签
'''
#pv
basedir="/home/qiqq/q3dl/datalinan/taihuyuan_pv/"
with open("/home/qiqq/q3dl/datalinan/taihuyuan_pv/trainbat.txt" ,'r') as  f1:
    pvssource =[ i.strip("\n") for i in  f1.readlines()]

with open("/home/qiqq/q3dl/datalinan/taihuyuan_pv/train.txt" ,'r') as  f2:
    pvn =[ i.strip("\n") for i in  f2.readlines()]

pvvv=list(set(pvn).difference(set(pvssource))) # b中有而a中没有的
print(len(pvvv))
print(pvvv)

for i in pvvv:
    imgpa=os.path.join(basedir,'images',i+'.png')
    labelpa=os.path.join(basedir,'labels',i+'.png')

    os.remove(imgpa)
    os.remove(labelpa)

print("Wancheng")