Minor changes

Author: dec4234

src/generator/mod.rs

@@ -31,7 +31,7 @@ pub struct TargetGenerator {
 	pub object_manager: ObjectManager,
 	background_loader: BackgroundLoader,
 	coco_generator: Arc<Mutex<CocoGenerator>>,
-	config: TargetGeneratorConfig,
+	pub(crate) config: TargetGeneratorConfig,
 	resized_cache: Cache<String, DynamicImage>,
 }
 

src/generator/util.rs

@@ -1,9 +1,6 @@
 use crate::generator::error::GenerationError;
 use image::metadata::Orientation;
 use image::DynamicImage;
-use imageproc::drawing::Canvas;
-use imageproc::geometric_transformations::Interpolation;
-use std::cmp::max;
 
 /// The standard Pixels Per Meter value that is used to calculate the size of objects in pixels.
 /// In reality this value is dependent on the altitude of the drone and various properties of the

src/main.rs

@@ -5,10 +5,18 @@ mod objects;
 mod generator;
 
 use simple_logger::SimpleLogger;
-use log::debug;
+use log::{debug, LevelFilter};
+use crate::generator::TargetGenerator;
 
 fn main() {
-    SimpleLogger::new().init().unwrap();
+    SimpleLogger::new().with_level(LevelFilter::Debug).init().unwrap();
     debug!("Starting...");
 
+    let mut tg = TargetGenerator::new("output", "backgrounds", "objects", "output/annotations.json").unwrap();
+    tg.config.permit_duplicates = true;
+    tg.config.permit_collisions = false;
+    //tg.config.visualize_bboxes = true;
+    tg.generate_targets(500, ..6u32, "output").unwrap();
+
+    tg.close();
 }