search

amazon-science / siam-mot

SiamMOT: Siamese Multi-Object Tracking
Apache License 2.0
477 stars 61 forks source link

KeyError: 'categories' #46

Closed Zizzzzzzz closed 2 years ago

Zizzzzzzz commented 2 years ago

Hello，my problem is

Traceback (most recent call last):
  File "tools/train_net.py", line 130, in <module>
    main()
  File "tools/train_net.py", line 126, in main
    train(cfg, train_dir, args.local_rank, args.distributed, logger)
  File "tools/train_net.py", line 69, in train
    start_iter=arguments["iteration"],
  File "/data/users/CHDHPC/2020224031/code/SiamMOT/siammot/data/build_train_data_loader.py", line 66, in build_train_data_loader
    dataset = build_dataset(cfg)
  File "/data/users/CHDHPC/2020224031/code/SiamMOT/siammot/data/build_train_data_loader.py", line 38, in build_dataset
    amodal=cfg.INPUT.AMODAL)
  File "/data/users/CHDHPC/2020224031/code/SiamMOT/siammot/data/image_dataset.py", line 43, in __init__
    self._det_classes = [c['name'] for c in self.dataset.loadCats(self.dataset.getCatIds())]
  File "/data/users/CHDHPC/2020224031/anaconda3/envs/FairMOT/lib/python3.6/site-packages/pycocotools/coco.py", line 168, in getCatIds
    cats = self.dataset['categories']
KeyError: 'categories'

can you tell me what should i do?