AI Label
This commit is contained in:
Semi212 committed
1 parent
897af7ce56
commit
ae53e9eac5
6 files changed
+177
-65
No files matched your search
@@ -7,28 +7,28 @@
|
||||
import os,torch, threading,argparse
|
||||
class global_para():
|
||||
#check()
|
||||
parser1 = argparse.ArgumentParser(description="CUI")
|
||||
parser1.add_argument("video_num", default='/data1/Hatcher/model/project/data_MPKFNet/rawdata/竹林嘴-3月3日候鸟视频.mp4', type=str,
|
||||
help='Video name.')
|
||||
parser1.add_argument("GPU", default='2', type=str,
|
||||
help='GPU.')
|
||||
parser1.add_argument("--thresh", default=0.4, type=float,
|
||||
help='GPU.')
|
||||
parser1.add_argument("--red", default=0, type=int,
|
||||
help='red.')
|
||||
parser1.add_argument("--add", default=0, type=str,
|
||||
help='add.')
|
||||
|
||||
args = parser1.parse_args()
|
||||
# parser1 = argparse.ArgumentParser(description="CUI")
|
||||
# parser1.add_argument("video_num", default='/data1/Hatcher/model/project/data_MPKFNet/rawdata/竹林嘴-3月3日候鸟视频.mp4', type=str,
|
||||
# help='Video name.')
|
||||
# parser1.add_argument("GPU", default='2', type=str,
|
||||
# help='GPU.')
|
||||
# parser1.add_argument("--thresh", default=0.4, type=float,
|
||||
# help='GPU.')
|
||||
# parser1.add_argument("--red", default=0, type=int,
|
||||
# help='red.')
|
||||
# parser1.add_argument("--add", default=0, type=str,
|
||||
# help='add.')
|
||||
#
|
||||
# args = parser1.parse_args()
|
||||
#print(args)
|
||||
root = "/home/gyd/output1"
|
||||
# os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
||||
os.environ['CUDA_VISIBLE_DEVICES'] = args.GPU
|
||||
devices = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
||||
camera_address = args.add
|
||||
VideoPath = args.video_num
|
||||
red_flag = args.red
|
||||
detect_thresh = args.thresh
|
||||
# root = "/home/gyd/output1"
|
||||
# # os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
||||
# os.environ['CUDA_VISIBLE_DEVICES'] = args.GPU
|
||||
# devices = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
||||
# camera_address = args.add
|
||||
# VideoPath = args.video_num
|
||||
# red_flag = args.red
|
||||
# detect_thresh = args.thresh
|
||||
|
||||
#fine_thresh2 = torch.tensor(fine_thresh2).cuda()
|
||||
birdIndex = 99999
|
||||
@@ -51,7 +51,7 @@ class global_para():
|
||||
|
||||
showThresh = 25
|
||||
destory_image = False
|
||||
Camera_device = args.video_num # ÉãÏñÍ·±àºÅ£¬0001
|
||||
# Camera_device = args.video_num # ÉãÏñÍ·±àºÅ£¬0001
|
||||
image_time = None # µ±Ç°´¦ÀíµÄ30ÊÓƵµÄ±àºÅ
|
||||
ImgDealQueue = [] # 750Ö¡ÊÓƵ£¬»º´æ¿Õ¼ä
|
||||
img_show = None
|
||||
@@ -80,28 +80,28 @@ class global_para():
|
||||
|
||||
# DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/exp21/weights/best.pt"
|
||||
# FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-JXPYH-Bird-3-4-20220429/_42911_PYH-3/best.pth"
|
||||
DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/KunMingCuiHu-Fish2/weights/best.pt"
|
||||
FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-KMDC-Fish-5-1-20220505/_5510_DianChiFish-4/best.pth"
|
||||
# DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/BeijingCuiHu-Bird26/weights/best.pt"
|
||||
# FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-cui90_1model0.7/_42111_CUI-94-1/best.pth"
|
||||
#DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/KunMingCuiHu-Fish2/weights/best.pt"
|
||||
#FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-KMDC-Fish-5-1-20220505/_5510_DianChiFish-4/best.pth"
|
||||
DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/BeijingCuiHu-Bird26/weights/best.pt"
|
||||
FineWeight = "/data/Hatcher/data/weigth/areaproject/VIT-BJCH_Bird_94_DDP/_5914_CUI-94-1/best.pth"
|
||||
#BHG_chineseName = ["kong", '白鹤', '白头鹤','白枕鹤']
|
||||
BHG_chineseName = ["kong","鲫鱼","麦穗鱼","鳑鲏","鲈鱼"]
|
||||
#BHG_chineseName = ["kong","鲫鱼","麦穗鱼","鳑鲏","鲈鱼"]
|
||||
|
||||
# BHG_chineseName = ['Kong', '白额雁', '白骨顶', '苍鹭', '苍鹭', '白秋沙鸭', '白腰草鹬', '斑头雁', '斑嘴鸭', '苍鹭', '赤膀鸭', '赤颈鸭', '苍鹭', '大麻鳽',
|
||||
# '豆雁', '反嘴鹬', '凤头鸊鷉', '凤头麦鸡', '凤头潜鸭', '黑鹳', '黑水鸡', '红嘴鸥', '鸿雁', '灰鹤', '灰雁', '绿翅鸭', '绿头鸭', '琵嘴鸭',
|
||||
# '普通鸬鹚', '普通秋沙鸭', '翘鼻麻鸭', '鹊鸭', '扇尾沙锥', '小白额雁', '小天鹅', '针尾鸭', '苍鹭', '暗绿绣眼鸟', '八哥', '白腹鹞', '白鹡鸰',
|
||||
# '白眼潜鸭', '草鹭', '池鹭', '戴胜', '黑翅长脚鹬', '黑鸢', '红隼', '花脸鸭', '环颈雉', '灰头绿啄木鸟', '灰头麦鸡', '矶鹬', '栗苇鳽',
|
||||
# '罗纹鸭', '牛背鹭', '普通翠鸟', '雀鹰', '喜鹊', '苍鹭', '鸳鸯', '棕背伯劳', '红胸秋沙鸭', '普通燕鸥', '小白额燕鸥', '丘鹬', '凤头蜂鹰',
|
||||
# '大天鹅', '大鸨', '小田鸡', '山斑鸠', '帆背潜鸭', '日本松雀鹰', '普通鵟', '松雀鹰', '毛脚鵟', '灰斑鸠', '白头鹞', '白尾鹞', '白眉鸭',
|
||||
# '白鹈鹕', '红头潜鸭', '苍鹰', '蒙古银鸥', '西方秧鸡', '赤嘴潜鸭', '阔嘴鹬', '雕鸮', '鹌鹑', '鹗', '黄苇鳽', '黑颈鹤', '小䴙䴘', '赤麻鸭']
|
||||
BHG_chineseName = ['Kong', '白额雁', '白骨顶', '苍鹭', '苍鹭', '白秋沙鸭', '白腰草鹬', '斑头雁', '斑嘴鸭', '苍鹭', '赤膀鸭', '赤颈鸭', '苍鹭', '大麻鳽',
|
||||
'豆雁', '反嘴鹬', '凤头鸊鷉', '凤头麦鸡', '凤头潜鸭', '黑鹳', '黑水鸡', '红嘴鸥', '鸿雁', '灰鹤', '灰雁', '绿翅鸭', '绿头鸭', '琵嘴鸭',
|
||||
'普通鸬鹚', '普通秋沙鸭', '翘鼻麻鸭', '鹊鸭', '扇尾沙锥', '小白额雁', '小天鹅', '针尾鸭', '苍鹭', '暗绿绣眼鸟', '八哥', '白腹鹞', '白鹡鸰',
|
||||
'白眼潜鸭', '草鹭', '池鹭', '戴胜', '黑翅长脚鹬', '黑鸢', '红隼', '花脸鸭', '环颈雉', '灰头绿啄木鸟', '灰头麦鸡', '矶鹬', '栗苇鳽',
|
||||
'罗纹鸭', '牛背鹭', '普通翠鸟', '雀鹰', '喜鹊', '苍鹭', '鸳鸯', '棕背伯劳', '红胸秋沙鸭', '普通燕鸥', '小白额燕鸥', '丘鹬', '凤头蜂鹰',
|
||||
'大天鹅', '大鸨', '小田鸡', '山斑鸠', '帆背潜鸭', '日本松雀鹰', '普通鵟', '松雀鹰', '毛脚鵟', '灰斑鸠', '白头鹞', '白尾鹞', '白眉鸭',
|
||||
'白鹈鹕', '红头潜鸭', '苍鹰', '蒙古银鸥', '西方秧鸡', '赤嘴潜鸭', '阔嘴鹬', '雕鸮', '鹌鹑', '鹗', '黄苇鳽', '黑颈鹤', '小䴙䴘', '赤麻鸭']
|
||||
# BHG_chineseName =["kong","绿头鸭","斑嘴鸭","灰雁","鸿雁","kong","小鷿鷈"]
|
||||
spiec_number_old = [0 for i in range(len(BHG_chineseName))]
|
||||
FineClassNum = 5
|
||||
FineClassNum = 94
|
||||
color_show = (255, 255, 255)
|
||||
color_box = (255, 0, 0)
|
||||
NoneLabel = "鱼"#"鹤类" # "鱼"
|
||||
fine_thresh1 = 6000
|
||||
fine_thresh2 = 6000
|
||||
NoneLabel = None#"鹤类" # "鱼"
|
||||
fine_thresh1 = 10000
|
||||
fine_thresh2 = 10000
|
||||
ShowAllNumFlag = True
|
||||
fontScale = 1
|
||||
"""
|
||||
|
||||
Reference in new issue
Block a user