其实这个是处理标签的:
数据集可以支持子文件夹
判断标签id是否数字可以改进一下:
str_1 = "123"
str_2 = "Abc"
str_3 = "123Abc"
# 用isdigit函数判断是否数字
print(str_1.isdigit())
print(str_2.isdigit())
print(str_3.isdigit())
def is_number(s):
try:
float(s)
return True
except ValueError:
pass
try:
import unicodedata
unicodedata.numeric(s)
return True
except (TypeError, ValueError):
pass
return False
@staticmethod
def id(file_path):
file_path=os.path.dirname(file_path)
if is_number(os.path.basename(os.path.dirname(file_path))):
id = int(os.path.basename(os.path.dirname(file_path)))
else:
id=int(os.path.basename(file_path))
return id