add yolo v10 and modify pipeline

2025-03-28 13:19:54 +08:00
parent 183299c06b
commit 798c596acc
471 changed files with 19109 additions and 7342 deletions
--- a/contrast/feat_extract/inference.py
+++ b/contrast/feat_extract/inference.py
@ -48,7 +48,7 @@ class FeatsInterface:
        modpath = os.path.join(curpath, conf.test_model)
        self.model.load_state_dict(torch.load(modpath, map_location=conf.device))
        self.model.eval()
-        print('load model {} '.format(conf.testbackbone))
+        # print('load model {} '.format(conf.testbackbone))
    
    def inference(self, images, detections=None):
        '''