Add code of GaitBase (#115)

* add resnet9 backbone and regular da ops

* add gait3d config

* fix invalid path CASIA-B* in windows

* add gaitbase config for all datasets

* rm unused OpenGait transform
This commit is contained in:
Junhao Liang
2023-03-20 14:59:08 +08:00
committed by GitHub
parent 9b74b39f80
commit 638c657763
16 changed files with 656 additions and 161 deletions
+1 -1
View File
@@ -231,7 +231,7 @@ def evaluate_segmentation(data, dataset):
return {"scalar/test_accuracy/mIOU": miou}
def evaluate_Gait3D(data, conf, metric='euc'):
def evaluate_Gait3D(data, dataset, metric='euc'):
msg_mgr = get_msg_mgr()
features, labels, cams, time_seqs = data['embeddings'], data['labels'], data['types'], data['views']