mirrors
/
DINOv
mirror of https://github.com/UX-Decoder/DINOv.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update train_net.py

main
Feng Li 2024-04-08 17:44:17 +08:00 committed by GitHub
parent 7ee80e88cf
commit 53bf20d5cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_net.py
View File

@ -476,7 +476,7 @@ class Trainer(DefaultTrainer):
print(idx)
@classmethod
def test_visual_openset(cls, cfg, model, eval_visual_openset_combine, evaluators=None):
def test_visual_openset(cls, cfg, model, evaluators=None):
# build dataloade
dataloaders = cls.build_test_loader(cfg, dataset_name=None)
dataset_names = cfg['DATASETS']['TEST']
@ -613,7 +613,7 @@ def main(args=None):
cfg.MODEL.WEIGHTS, resume=args.resume
)
if args.eval_visual_openset:
res = Trainer.test_visual_openset(cfg, model, args.eval_visual_openset_combine)
res = Trainer.test_visual_openset(cfg, model)
elif args.eval_track_prev:
res = Trainer.test_tracking_prev(cfg, model)
elif args.eval_get_content_features: