-
Notifications
You must be signed in to change notification settings - Fork 1.2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add option to render evaluated images #2135
Conversation
@@ -173,7 +175,7 @@ def get_eval_image_metrics_and_images(self, step: int): | |||
|
|||
@abstractmethod | |||
@profiler.time_function | |||
def get_average_eval_image_metrics(self, step: Optional[int] = None): | |||
def get_average_eval_image_metrics(self, step: Optional[int] = None, output_path: Optional[Path]=None): | |||
"""Iterate over all the images in the eval dataset and get the average.""" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Add doc for Args
@@ -339,7 +341,7 @@ def get_eval_image_metrics_and_images(self, step: int): | |||
return metrics_dict, images_dict | |||
|
|||
@profiler.time_function | |||
def get_average_eval_image_metrics(self, step: Optional[int] = None): | |||
def get_average_eval_image_metrics(self, step: Optional[int] = None, output_path: Optional[Path] = None): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Add doc for Args
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Head branch was pushed to by a user without write access
sorry for the delay here @tancik - this is now passing the build |
Adding an optional "render_output_path" parameter to the eval script that saves the rendered image outputs to the specified path.