CL will fix the typos in the tasks files
PiperOrigin-RevId: 522240681
This commit is contained in:
committed by
Copybara-Service
parent
d5def9e24d
commit
7ae4d0175a
@@ -84,7 +84,7 @@ class Dataset(object):
|
||||
create randomness during model training.
|
||||
preprocess: A function taking three arguments in order, feature, label and
|
||||
boolean is_training.
|
||||
drop_remainder: boolean, whether the finaly batch drops remainder.
|
||||
drop_remainder: boolean, whether the finally batch drops remainder.
|
||||
|
||||
Returns:
|
||||
A TF dataset ready to be consumed by Keras model.
|
||||
|
||||
@@ -32,7 +32,7 @@ class BaseHParams:
|
||||
epochs: Number of training iterations over the dataset.
|
||||
steps_per_epoch: An optional integer indicate the number of training steps
|
||||
per epoch. If not set, the training pipeline calculates the default steps
|
||||
per epoch as the training dataset size devided by batch size.
|
||||
per epoch as the training dataset size divided by batch size.
|
||||
shuffle: True if the dataset is shuffled before training.
|
||||
export_dir: The location of the model checkpoint files.
|
||||
distribution_strategy: A string specifying which Distribution Strategy to
|
||||
|
||||
@@ -21,7 +21,7 @@ package(
|
||||
default_visibility = ["//mediapipe:__subpackages__"],
|
||||
)
|
||||
|
||||
# TODO: Remove the unncessary test data once the demo data are moved to an open-sourced
|
||||
# TODO: Remove the unnecessary test data once the demo data are moved to an open-sourced
|
||||
# directory.
|
||||
filegroup(
|
||||
name = "testdata",
|
||||
|
||||
@@ -155,8 +155,8 @@ class Dataset(classification_dataset.ClassificationDataset):
|
||||
ObjectDetectorDataset object.
|
||||
"""
|
||||
# Get TFRecord Files
|
||||
tfrecord_file_patten = cache_prefix + '*.tfrecord'
|
||||
matched_files = tf.io.gfile.glob(tfrecord_file_patten)
|
||||
tfrecord_file_pattern = cache_prefix + '*.tfrecord'
|
||||
matched_files = tf.io.gfile.glob(tfrecord_file_pattern)
|
||||
if not matched_files:
|
||||
raise ValueError('TFRecord files are empty.')
|
||||
|
||||
|
||||
@@ -345,7 +345,7 @@ def _coco_annotations_to_lists(
|
||||
Args:
|
||||
bbox_annotations: List of dicts with keys ['bbox', 'category_id']
|
||||
image_height: Height of image
|
||||
image_width: Width of iamge
|
||||
image_width: Width of image
|
||||
|
||||
Returns:
|
||||
(data, num_annotations_skipped) tuple where data contains the keys:
|
||||
|
||||
Reference in New Issue
Block a user