#coding:utf-8

# 图片及视频检测结果保存路径
save_path = 'save_data'

# 使用的模型路径
# model_path = 'models/best.pt'

# names = {0: 'Rescued', 1: 'Rescuing'}

# CH_names = ['已救援', '救援中']


# 猫和狗
# 训练好的模型路径
# model_path = 'runs/detect/train5/weights/best.pt'
# # 类别名称
# names = {0: 'cat', 1: 'dog'}
# # 中文类别名称
# CH_names = ['猫', '狗']


# # 马甲
# # 训练好的模型路径
# model_path = 'runs/detect/train5/weights/best.pt'
# # # 类别名称
# names = {0: 'vest', 1: 'novest'}
# # # 中文类别名称
# CH_names = ['有马甲', '无马甲']

# # 仪表盘
# # 训练好的模型路径
# model_path = 'runs/detect/train6/weights/best.pt'
# # # 类别名称
# names = {0: 'dashboard', 1: 'nodashboard'}
# # # 中文类别名称
# CH_names = ['仪表盘', '无仪表盘']


# # 搭电设备
# # 训练好的模型路径
# model_path = 'runs/detect/train9/weights/best.pt'
# # # 类别名称
# names = {0: 'JumperCable', 1: 'NoJumperCable'}
# # # 中文类别名称
# CH_names = ['有搭电设备', '无搭电设备']


# 搭电测试
# 训练好的模型路径
model_path = 'runs/segment/train11/weights/best.pt'
# # 类别名称
names = {0: 'da_dian_xian', 1: 'tuo_che_shang_you_che',2:'yi_biao_pan',3:'lun_tai',4:'ping_an_ma_jia',5:'jian_ce_yi'}

# # 中文类别名称
CH_names = ['搭电线','拖车上油车','仪表盘','轮台','平安马甲','检测仪']