Untitled

mail@pastecode.io avatar
unknown
python
a year ago
420 B
0
Indexable
def listup_tfrecord_targets_csv(src_path, dst_path, dst_fn, target_drv_ids=[]):
	dst_fn_path=os.path.join(dst_path, dst_fn)
	target='VG'
	
	db_dct={}
	heads = [some column names]

	for hd in heads:
		db_dct[hd]=[]
	db_dct['Image Path']=[]
	db_dct['GT Path']=[]
	for drv_dn in oslistdir(src_path):
		drv_dir=os.path.join(src_path,drv_dn,'GT_data')
		if not os.path.isdir(drv_idr):
			print("wrong directory")
			continue