修正了一个环境错误

Author: vvEverett

main.py

@@ -6,8 +6,8 @@
 from nanodet.util import Logger, cfg, load_config, load_model_weight
 
 VIDEO_FILE = r"D:\shijue\LiquidDrop\22.avi"
-WEIGHTS_PATH = r'D:\shijue\multi-object-tracker\weight\LiquidV4.pth'
-CONFIG_FILE_PATH = r'D:\shijue\multi-object-tracker\config\LiquidDetect416.yml'
+WEIGHTS_PATH = 'weight/LiquidV4.pth'
+CONFIG_FILE_PATH = 'config/LiquidDetect416.yml'
 
 tracker = CentroidTracker(max_lost=0, tracker_output_format='mot_challenge')
 

nanodet/data/collate.py

@@ -16,7 +16,9 @@
 import re
 
 import torch
-from torch._six import string_classes
+# from torch._six import string_classes
+
+string_classes = (str, bytes)
 
 np_str_obj_array_pattern = re.compile(r"[SaUO]")
 

requirements.txt

@@ -7,7 +7,7 @@ onnx-simplifier
 opencv-python
 pyaml
 pycocotools
-pytorch-lightning>=1.7.0
+pytorch-lightning==1.7.0
 tabulate
 tensorboard
 termcolor