61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
#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 = ['搭电线','拖车上油车','仪表盘','轮台','平安马甲','检测仪']
|
|
|
|
|
|
|
|
|