Skip to content

Commit

Permalink
fix get_train_dataset_pipline to get_train_dataset_pipeline(I think i…
Browse files Browse the repository at this point in the history
…t is mistyping)
  • Loading branch information
tucan9389 committed Jul 25, 2018
1 parent 93f4b94 commit 89edf1c
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
*/.DS_Store
training/src/__pycache__
.idea/*
4 changes: 2 additions & 2 deletions training/src/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,5 +97,5 @@ def _get_dataset_pipline(json_filename, batch_size, epoch, buffer_size):
return dataset


def get_train_dataset_pipline(batch_size=32, epoch=10, buffer_size=1):
return _get_dataset_pipline(TRAIN_JSON, batch_size, epoch, buffer_size, )
def get_train_dataset_pipeline(batch_size=32, epoch=10, buffer_size=1):
return _get_dataset_pipeline(TRAIN_JSON, batch_size, epoch, buffer_size, )
4 changes: 2 additions & 2 deletions training/src/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,14 +22,14 @@

from datetime import datetime

from dataset import get_train_dataset_pipline
from dataset import get_train_dataset_pipeline
from networks import get_network
from dataset_prepare import CocoPose
from dataset_augment import set_network_input_wh, set_network_scale


def get_train_input(batchsize, epoch):
train_ds = get_train_dataset_pipline(batch_size=batchsize, epoch=epoch, buffer_size=100)
train_ds = get_train_dataset_pipeline(batch_size=batchsize, epoch=epoch, buffer_size=100)
iter = train_ds.make_one_shot_iterator()
_ = iter.get_next()
return _[0], _[1]
Expand Down

0 comments on commit 89edf1c

Please sign in to comment.