diff --git a/research/cv/hed/src/dataset.py b/research/cv/hed/src/dataset.py
index 842d1438d851621426976f7cf95dd54931f6727d..74233afd40c168dbdb2624ca509219aeba53df22 100644
--- a/research/cv/hed/src/dataset.py
+++ b/research/cv/hed/src/dataset.py
@@ -100,7 +100,6 @@ def create_dataset(data_path, is_training=True, is_shuffle=True, batch_size=1, r
         dataset = HED_Dataset_e(data_path, is_training)
         dataloader = ds.GeneratorDataset(dataset, ['test', 'label'], num_parallel_workers=8, shuffle=is_shuffle)
         dataloader = dataloader.map(input_columns='test', operations=C.TypeCast(mindspore.float32))
-        dataloader = dataloader.map(input_columns='label', operations=C.TypeCast(mindspore.float32))
         dataloader = dataloader.batch(batch_size, drop_remainder=False)
 
     # apply DatasetOps