Skip to content
Snippets Groups Projects
Unverified Commit 0da4a5b1 authored by Daniel Ecer's avatar Daniel Ecer Committed by GitHub
Browse files

bounding box related file structure (#405)

* moved bbox main to separate module

* renamed test module to match main module under test
parent bf2012e2
No related branches found
No related tags found
No related merge requests found
import argparse
import logging
from typing import List, Optional
from sciencebeam_gym.tools.image_annotation.find_bounding_boxes_utils import (
FindBoundingBoxPipelineFactory,
parse_args,
process_args
)
LOGGER = logging.getLogger(__name__)
def run(args: argparse.Namespace):
FindBoundingBoxPipelineFactory(args).run(
args
)
def main(argv: Optional[List[str]] = None):
args = parse_args(argv)
if args.debug:
for name in ['__main__', 'sciencebeam_gym']:
logging.getLogger(name).setLevel(logging.DEBUG)
LOGGER.info('args: %s', args)
process_args(args)
run(args)
from sciencebeam_gym.tools.image_annotation.find_bounding_boxes_utils import main
if __name__ == '__main__':
......
......@@ -28,14 +28,11 @@ from sciencebeam_gym.tools.image_annotation.find_bounding_boxes_utils import (
FindBoundingBoxPipelineFactory,
GraphicImageNotFoundError,
format_coords_attribute_value,
main,
parse_args,
save_annotated_images
)
from sciencebeam_gym.tools.image_annotation.find_bounding_boxes import (
main
)
LOGGER = logging.getLogger(__name__)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment