(age_gender) fix incorrect args (#222)

tue-robotics · Mar 12, 2024 · bdce167 · bdce167
2 parents 283c302 + 9b9d7f5
commit bdce167
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/image_recognition_age_gender/scripts/face_properties_node b/image_recognition_age_gender/scripts/face_properties_node
@@ -12,7 +12,7 @@ from image_recognition_util import image_writer
 
 
 class FacePropertiesNode:
-    def __init__(self, weights_file_path, img_size, save_images_folder, use_gpu):
+    def __init__(self, weights_file_path: str, img_size: int, save_images_folder: bool, use_gpu: bool):
         """
         ROS node that wraps the PyTorch age gender estimator
         """
@@ -91,7 +91,7 @@ if __name__ == '__main__':
         sys.exit(1)
 
     try:
-        FacePropertiesNode(weights_file_path, img_size, depth, width, save_images_folder, use_gpu)
+        FacePropertiesNode(weights_file_path, img_size, save_images_folder, use_gpu)
         updater = diagnostic_updater.Updater()
         updater.setHardwareID("none")
         updater.add(diagnostic_updater.Heartbeat())