clean_code

This commit is contained in:
shenwei41 2021-07-13 10:46:07 +08:00
parent 897ab2fe01
commit 025ad59f62
4 changed files with 3 additions and 1 deletions

View File

@ -27,7 +27,6 @@
namespace mindspore {
namespace dataset {
// Fetches one row of data from the iterator as a column map.
Status DatasetIterator::GetNextAsMap(TensorMap *out_map) {
if (out_map == nullptr) {

View File

@ -516,6 +516,7 @@ def check_save_model(method):
return new_method
def check_sentence_piece_tokenizer(method):
"""A wrapper that wraps a parameter checker to the original function."""

View File

@ -1291,6 +1291,7 @@ def rgb_to_bgrs(np_rgb_imgs, is_hwc):
return rgb_to_bgr(np_rgb_imgs, is_hwc)
return np.array([rgb_to_bgr(img, is_hwc) for img in np_rgb_imgs])
def rgb_to_hsv(np_rgb_img, is_hwc):
"""
Convert RGB img to HSV img.

View File

@ -279,6 +279,7 @@ class DatasetHelper:
"""Get shape range(min shape, max shape) of dynamic data."""
return self.iter.dynamic_min_max_shapes()
class _DatasetIter:
"""Base iter for dataset helper"""