TrTr icon indicating copy to clipboard operation
TrTr copied to clipboard

Published 20 hours ago • verified publisher icon tongtybj

Training Problem

Open DavidZhangdw opened this issue 3 years ago • 8 comments

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.

DavidZhangdw avatar Jul 27 '21 01:07 DavidZhangdw

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

tongtybj avatar Jul 27 '21 01:07 tongtybj

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

tongtybj avatar Jul 27 '21 01:07 tongtybj

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

tongtybj avatar Jul 27 '21 02:07 tongtybj

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~

DavidZhangdw avatar Jul 27 '21 02:07 DavidZhangdw

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).

tongtybj avatar Jul 27 '21 02:07 tongtybj

Thank you, I have a try.

DavidZhangdw avatar Jul 27 '21 02:07 DavidZhangdw

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

DavidZhangdw avatar Jul 28 '21 09:07 DavidZhangdw

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

tongtybj avatar Sep 18 '21 11:09 tongtybj