BigGait_v3

2024-05-11 11:23:06 +08:00
parent 2f3bb9af35
commit 762247c34a
3 changed files with 3 additions and 1 deletions
@@ -19,7 +19,7 @@ from kornia import morphology as morph
 import random

 # import GaitBase & DINOv2_small
-from .BigGait_utils.GaitBase import Baseline
+from .BigGait_utils.BigGait_GaitBase import Baseline
 from .BigGait_utils.DINOv2 import vit_small
 from .BigGait_utils.save_img import save_image, pca_image