diff --git a/image_recognition_age_gender/scripts/face_properties_node b/image_recognition_age_gender/scripts/face_properties_node index 28169406..89e4d0d3 100755 --- 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())