search

tongtybj / TrTr

TrTr: Visual Tracking with Transformer
79 stars 13 forks source link

Training Problem #9

Open DavidZhangdw opened 3 years ago

DavidZhangdw commented 3 years ago

image

When I run main.py to excute training, Rasing this problem:

TypeError: can not load module pickle, currently can only support json and picke.

But I have generated train/val.Json and train/val.pickle files.

tongtybj commented 3 years ago

The error is KeyError of frame_size. How did you generate your Json or pickle files?

tongtybj commented 3 years ago

And which dataset did you use? VID, YoutubeBB, LaSOT, etc?

tongtybj commented 3 years ago

Can you share your the annotation files (.json or .pickle) to me via email?

DavidZhangdw commented 3 years ago

I generate my Json or pickle files using the methods in PySOT framework. My training datasets contains VID, Coco, YT-BB, LaSOT and GOT-10k. OK, I will sent the json and pickle files to you via email. Thanks~

tongtybj commented 3 years ago

Oh, I see.

The problem is we have a different format for recording the annotation dataset from PySOT framework. For example frame_size is a new key which is not included in PySOT one.

I recommend you to follow https://github.com/tongtybj/TrTr/blob/master/datasets/README.md , and the README.md in each dataset to generate the .json or .pickle (default is .pickle).

DavidZhangdw commented 3 years ago

Thank you, I have a try.

DavidZhangdw commented 3 years ago

Can you share your .json and .pickle files of youtube-bb datasets for me via email ?

tongtybj commented 3 years ago

Sorry for the late response. I uploaded the curated yt_bb dataset in google drive. Please follow the instruction in following link: https://github.com/tongtybj/TrTr/blob/master/datasets/yt_bb/README.md#method2-download-curated-dataset-from-google-drive