update bug

data_generator.py

@@ -24,9 +24,12 @@ class DataGenerator:
         self.min_cam_table_included_degree = config["runner"]["generate"]["min_cam_table_included_degree"]
         self.random_view_ratio = config["runner"]["generate"]["random_view_ratio"]
         self.binocular_vision = config["runner"]["generate"]["binocular_vision"]
-        self.set_status_path = "http://localhost:5000/project/set_status"
-        self.log_path = "http://localhost:5000/project/add_log"
-        self.obj_name_list = os.listdir(self.obj_dir)
+        self.port = config["runner"]["generate"]["port"]
+        self.from_idx = config["runner"]["generate"]["from"]
+        self.to_idx = config["runner"]["generate"]["to"]
+        self.set_status_path = f"http://localhost:{self.port}/project/set_status"
+        self.log_path = f"http://localhost:{self.port}/project/add_log"
+        self.obj_name_list = os.listdir(self.obj_dir)[self.from_idx: self.to_idx]
         self.target_obj = None
         self.stopped = False
         self.random_obj_list = []
@@ -82,9 +85,6 @@ class DataGenerator:
         
         height = random.uniform(config["min_height"], config["max_height"])
         radius = random.uniform(config["min_radius"], config["max_radius"])
-        R = random.uniform(config["min_R"], config["max_R"])
-        G = random.uniform(config["min_G"], config["max_G"])
-        B = random.uniform(config["min_B"], config["max_B"])
         while height > 0.5 * radius:
             height = random.uniform(config["min_height"], config["max_height"])
 
@@ -112,9 +112,6 @@ class DataGenerator:
         self.display_table_config = {
             "height": height,
             "radius": radius,
-            "R": R,
-            "G": G,
-            "B": B,
             "location": list(platform.location)
         }
         return platform