add annotation to original gtsrb dataloader.

5 years ago · 0894e4061b
1 changed files with 2 additions and 1 deletions
--- a/datasets/gtsrb.py
+++ b/datasets/gtsrb.py
@ -1,4 +1,5 @@
 """Dataset setting and data loader for GTSRB."""
 """Dataset setting and data loader for GTSRB. Raw format and not use roi info.
 """
 import os
 import torch