Panoramic

common.py

@@ -8,9 +8,9 @@ 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/PYH-DEMO.mp4', type=str,
+    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='6', type=str,
+    parser1.add_argument("GPU", default='2', type=str,
                         help='GPU.')
     parser1.add_argument("--thresh", default=0.4, type=float,
                         help='GPU.')
@@ -79,13 +79,13 @@ class global_para():
     Fine_thresh = 0.4
 
     # DetWeight = "/data1/Hatcher/model/Detection/yolov5/runs/train/exp21/weights/best.pt"
-    # FineWeight = "/data/Hatcher/data/weigth/areaproject/resnet50-PYH-3-20.7/_42522_PYH-3/best.pth"
+    # 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-DianChiFish-3-3-0.7/_4289_DianChiFish-4/best.pth"
+    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"
-    # BHG_chineseName = ["kong", '白鹤', '白头鹤','白枕鹤']#["kong","鲫鱼","麦穗鱼","鳑鲏"]
-    BHG_chineseName = ["kong","鲫鱼","麦穗鱼","鳑鲏"]
+    #BHG_chineseName = ["kong", '白鹤', '白头鹤','白枕鹤']
+    BHG_chineseName = ["kong","鲫鱼","麦穗鱼","鳑鲏","鲈鱼"]
 
     # BHG_chineseName = ['Kong', '白额雁', '白骨顶', '苍鹭', '苍鹭', '白秋沙鸭', '白腰草鹬', '斑头雁', '斑嘴鸭', '苍鹭', '赤膀鸭', '赤颈鸭', '苍鹭', '大麻鳽',
     #                    '豆雁', '反嘴鹬', '凤头鸊鷉', '凤头麦鸡', '凤头潜鸭', '黑鹳', '黑水鸡', '红嘴鸥', '鸿雁', '灰鹤', '灰雁', '绿翅鸭', '绿头鸭', '琵嘴鸭',
@@ -96,12 +96,13 @@ class global_para():
     #                    '白鹈鹕', '红头潜鸭', '苍鹰', '蒙古银鸥', '西方秧鸡', '赤嘴潜鸭', '阔嘴鹬', '雕鸮', '鹌鹑', '鹗', '黄苇鳽', '黑颈鹤', '小䴙䴘', '赤麻鸭']
     # BHG_chineseName =["kong","绿头鸭","斑嘴鸭","灰雁","鸿雁","kong","小鷿鷈"]
     spiec_number_old = [0 for i in range(len(BHG_chineseName))]
-    FineClassNum = 4
+    FineClassNum = 5
     color_show = (255, 255, 255)
     color_box = (255, 0, 0)
-    NoneLabel = "鱼"  # "鹤类"
+    NoneLabel = "鱼"#"鹤类" # "鱼"
     fine_thresh1 = 6000
-    fine_thresh2 = 10000
+    fine_thresh2 = 6000
+    ShowAllNumFlag = True
     fontScale = 1
 """
 1.change weight
@@ -113,6 +114,7 @@ class global_para():
 7. change main
 8. change fine_thresh2
 9. GPU
+10. change ShowAllNumFlag
 fontScale  plot file para to adjust chinese   400W ---2  
                                               200W ---1  
                                               800W ---3