Update voc.py

own
Lin Manhui 2 years ago committed by GitHub
parent 34d35ec27f
commit daf06b118d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      paddlers/datasets/voc.py

@ -289,7 +289,7 @@ class VOCDetDataset(BaseDataset):
self.num_max_boxes = max(self.num_max_boxes, len(objs)) self.num_max_boxes = max(self.num_max_boxes, len(objs))
if not ct: if not ct:
logging.error("No voc record found in %s' % (file_list)", exit=True) logging.error("No voc record found in %s ." % (file_list), exit=True)
self.pos_num = len(self.file_list) self.pos_num = len(self.file_list)
if self.allow_empty and neg_file_list: if self.allow_empty and neg_file_list:
self.file_list += self._sample_empty(neg_file_list) self.file_list += self._sample_empty(neg_file_list)

Loading…
Cancel
Save