forked from mindspore-Ecosystem/mindspore
clean_code
This commit is contained in:
parent
897ab2fe01
commit
025ad59f62
|
@ -27,7 +27,6 @@
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace dataset {
|
namespace dataset {
|
||||||
|
|
||||||
// Fetches one row of data from the iterator as a column map.
|
// Fetches one row of data from the iterator as a column map.
|
||||||
Status DatasetIterator::GetNextAsMap(TensorMap *out_map) {
|
Status DatasetIterator::GetNextAsMap(TensorMap *out_map) {
|
||||||
if (out_map == nullptr) {
|
if (out_map == nullptr) {
|
||||||
|
|
|
@ -516,6 +516,7 @@ def check_save_model(method):
|
||||||
|
|
||||||
return new_method
|
return new_method
|
||||||
|
|
||||||
|
|
||||||
def check_sentence_piece_tokenizer(method):
|
def check_sentence_piece_tokenizer(method):
|
||||||
|
|
||||||
"""A wrapper that wraps a parameter checker to the original function."""
|
"""A wrapper that wraps a parameter checker to the original function."""
|
||||||
|
|
|
@ -1291,6 +1291,7 @@ def rgb_to_bgrs(np_rgb_imgs, is_hwc):
|
||||||
return rgb_to_bgr(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])
|
return np.array([rgb_to_bgr(img, is_hwc) for img in np_rgb_imgs])
|
||||||
|
|
||||||
|
|
||||||
def rgb_to_hsv(np_rgb_img, is_hwc):
|
def rgb_to_hsv(np_rgb_img, is_hwc):
|
||||||
"""
|
"""
|
||||||
Convert RGB img to HSV img.
|
Convert RGB img to HSV img.
|
||||||
|
|
|
@ -279,6 +279,7 @@ class DatasetHelper:
|
||||||
"""Get shape range(min shape, max shape) of dynamic data."""
|
"""Get shape range(min shape, max shape) of dynamic data."""
|
||||||
return self.iter.dynamic_min_max_shapes()
|
return self.iter.dynamic_min_max_shapes()
|
||||||
|
|
||||||
|
|
||||||
class _DatasetIter:
|
class _DatasetIter:
|
||||||
"""Base iter for dataset helper"""
|
"""Base iter for dataset helper"""
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue