train.py
: The primary script for training models.losses.py
: Contains definitions for loss functions.models.py
: Contains the base class for defining a model.video_level_models.py
: Contains definitions for models that take aggregated features as input.frame_level_models.py
: Contains definitions for models that take frame- level features as input.model_util.py
: Contains functions that are of general utility for implementing models.export_model.py
: Provides a class to export a model during training for later use in batch prediction.readers.py
: Contains definitions for the Video dataset and Frame dataset readers.
eval.py
: The primary script for evaluating models.eval_util.py
: Provides a class that calculates all evaluation metrics.average_precision_calculator.py
: Functions for calculating average precision.mean_average_precision_calculator.py
: Functions for calculating mean average precision.
inference.py
: Generates an output file containing predictions of the model over a set of videos.