search

Johnathan-Xie / ZSD-YOLO

GNU General Public License v3.0
50 stars 2 forks source link

Dataset #1

Closed Daybreak-Zheng closed 1 year ago

Daybreak-Zheng commented 2 years ago

Hello, excellent work. May I ask how the label file in PT format in the training set is generated? Looking forward to your reply

Johnathan-Xie commented 1 year ago

Hi, these label files are generated when running torch.save() on the tensor which contains the class, bounding box, and embedding vectors. All data processing was performed via notebooks which I could share if you need to process a new dataset.