forked from mindspore-Ecosystem/mindspore
MD UT - Add missing set and restore seed.
test_places365_content_check - delete visualize_dataset
This commit is contained in:
parent
e17ba24394
commit
e66aba9ef5
|
@ -71,8 +71,7 @@ def test_places365_content_check():
|
|||
logger.info("Test Places365Dataset Op with content check")
|
||||
sampler = ds.SequentialSampler(num_samples=4)
|
||||
data1 = ds.Places365Dataset(dataset_dir=DATA_DIR, usage='val', small=True, decode=True, sampler=sampler)
|
||||
images, labels = load_places365(DATA_DIR)
|
||||
visualize_dataset(images, labels)
|
||||
_, labels = load_places365(DATA_DIR)
|
||||
num_iter = 0
|
||||
# in this example, each dictionary has keys "image" and "label"
|
||||
image_list, label_list = [], []
|
||||
|
|
|
@ -84,6 +84,8 @@ def test_random_posterize_op_fixed_point_c(plot=False, run_golden=True):
|
|||
Test RandomPosterize in C transformations with fixed point
|
||||
"""
|
||||
logger.info("test_random_posterize_op_c")
|
||||
original_seed = config_get_set_seed(55)
|
||||
original_num_parallel_workers = config_get_set_num_parallel_workers(1)
|
||||
|
||||
# define map operations
|
||||
transforms1 = [
|
||||
|
@ -115,6 +117,10 @@ def test_random_posterize_op_fixed_point_c(plot=False, run_golden=True):
|
|||
if plot:
|
||||
visualize_list(image_original, image_posterize)
|
||||
|
||||
# Restore configuration
|
||||
ds.config.set_seed(original_seed)
|
||||
ds.config.set_num_parallel_workers(original_num_parallel_workers)
|
||||
|
||||
|
||||
def test_random_posterize_default_c_md5(plot=False, run_golden=True):
|
||||
"""
|
||||
|
|
|
@ -76,6 +76,8 @@ def test_random_solarize_mnist(plot=False, run_golden=True):
|
|||
"""
|
||||
Test RandomSolarize op with MNIST dataset (Grayscale images)
|
||||
"""
|
||||
original_seed = config_get_set_seed(0)
|
||||
original_num_parallel_workers = config_get_set_num_parallel_workers(1)
|
||||
|
||||
mnist_1 = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
|
||||
mnist_2 = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
|
||||
|
@ -99,6 +101,9 @@ def test_random_solarize_mnist(plot=False, run_golden=True):
|
|||
filename = "random_solarize_02_result.npz"
|
||||
save_and_check_md5(mnist_2, filename, generate_golden=GENERATE_GOLDEN)
|
||||
|
||||
ds.config.set_seed(original_seed)
|
||||
ds.config.set_num_parallel_workers(original_num_parallel_workers)
|
||||
|
||||
|
||||
def test_random_solarize_errors():
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue