henry000 commited on
Commit
2028dc3
·
1 Parent(s): 3637d28

✏️ [Fix] a typo of dataset name, and remove warning

Browse files
utils/converter_json2txt.py CHANGED
@@ -25,6 +25,8 @@ def process_annotations(
25
  """
26
  for image_id, annotations in tqdm(image_annotations.items(), desc="Processing annotations"):
27
  file_path = os.path.join(output_dir, f"{image_id:0>12}.txt")
 
 
28
  with open(file_path, "w") as file:
29
  for annotation in annotations:
30
  process_annotation(annotation, image_info_dict[image_id], id_to_idx, file)
 
25
  """
26
  for image_id, annotations in tqdm(image_annotations.items(), desc="Processing annotations"):
27
  file_path = os.path.join(output_dir, f"{image_id:0>12}.txt")
28
+ if not annotations:
29
+ continue
30
  with open(file_path, "w") as file:
31
  for annotation in annotations:
32
  process_annotation(annotation, image_info_dict[image_id], id_to_idx, file)
utils/dataloader.py CHANGED
@@ -45,7 +45,7 @@ class YoloDataset(Dataset):
45
  if data is None:
46
  logger.info("Generating {} cache", phase_name)
47
  images_path = path.join(dataset_path, "images", phase_name)
48
- labels_path = path.join(dataset_path, "label", phase_name)
49
  data = self.filter_data(images_path, labels_path)
50
  cache[phase_name] = data
51
 
 
45
  if data is None:
46
  logger.info("Generating {} cache", phase_name)
47
  images_path = path.join(dataset_path, "images", phase_name)
48
+ labels_path = path.join(dataset_path, "labels", phase_name)
49
  data = self.filter_data(images_path, labels_path)
50
  cache[phase_name] = data
51