diff --git a/src/convert_sly_to_yolov5.py b/src/convert_sly_to_yolov5.py index 52dea91..262fc27 100644 --- a/src/convert_sly_to_yolov5.py +++ b/src/convert_sly_to_yolov5.py @@ -194,14 +194,14 @@ def _add_to_split(image_id, img_name, split_ids, split_image_paths, labels_dir, else: loop.run_until_complete(coro) + progress.iters_done_report(len(ids_to_download)) + if unsupported_shapes > 0: sly.logger.warning( f"Dataset {dataset.name}: " f"{unsupported_shapes} objects with unsupported geometry types have been {process_shapes_message}" ) - progress.iters_done_report(len(ids_to_download)) - data_yaml = { "train": "../{}/images/train".format(result_dir_name), "val": "../{}/images/val".format(result_dir_name),