语义分割数据集idd和cityscapes是很类似的,他们可以产生相同的id意义的ground truth, 代码在https://github.com/AutoNUE/public-code
可以设置和cs的类别意义一样的setting,idd根据json自动生成cs格式的gt
因为advent的代码是传入cityscapes的路径list,所以需要获取诸如一个文件夹下所有子文件中的路径,
即递归获取文件夹中子文件夹的所有文件的full path,用如下的code即可
import os
def listdir(path, list_name):
for file in os.listdir(path):
file_path = os.path.join(path, file)
if os.path.isdir(file_path):
listdir(file_path, list_name)
else:
temp = file_path.split('/')
temp0 = temp[-2]+'/'+temp[-1]
list_name.append(temp0)
list_name = []
path=filePath
listdir(path,list_name)
print(list_name)
如果要写到txt文件之中的话,需要以下代码
with open('./idd.txt','w') as f: #要存入的txt
write=''
for i in list_name:
write=write+str(i)+'\n'
f.write(write)
然后就可以得到诸如下面的txt文件了
正是和cityscapes一样格式的txt文件,也是我们想要的结果