Skip to content

Commit

Permalink
Change evaluate command to eval
Browse files Browse the repository at this point in the history
  • Loading branch information
Agustín Azzinnari authored and vierja committed Feb 8, 2018
1 parent 33caa50 commit 306a547
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 8 deletions.
4 changes: 2 additions & 2 deletions luminoth/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

import click

from luminoth.eval import evaluate
from luminoth.eval import eval
from luminoth.predict import predict
from luminoth.tools import cloud, dataset, server
from luminoth.train import train
Expand All @@ -28,6 +28,6 @@ def cli():
cli.add_command(cloud)
cli.add_command(dataset)
cli.add_command(predict)
cli.add_command(evaluate)
cli.add_command(eval)
cli.add_command(train)
cli.add_command(server)
8 changes: 2 additions & 6 deletions luminoth/eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@
@click.option('--files-per-class', type=int, default=10, help='How many files per class display in every epoch.') # noqa
@click.option('--iou-threshold', type=float, default=0.5, help='IoU threshold to use.') # noqa
@click.option('--min-probability', type=float, default=0.5, help='Minimum probability to use.') # noqa
def evaluate(dataset_split, config_files, watch, from_global_step,
override_params, files_per_class, iou_threshold, min_probability):
def eval(dataset_split, config_files, watch, from_global_step,
override_params, files_per_class, iou_threshold, min_probability):
"""Evaluate models using dataset."""

# If the config file is empty, our config will be the base_config for the
Expand Down Expand Up @@ -589,7 +589,3 @@ def calculate_map(output_per_batch, num_classes, iou_threshold=0.5):
mean_ap = np.mean(ap_per_class)

return mean_ap, ap_per_class


if __name__ == '__main__':
evaluate()

0 comments on commit 306a547

Please sign in to comment.