From 5df4a3c48ecf88b50b61f50740613d052473254f Mon Sep 17 00:00:00 2001
From: zhanghuiyao <1814619459@qq.com>
Date: Tue, 21 Jun 2022 19:00:14 +0800
Subject: [PATCH] fix hed eval dataset bug with master
---
research/cv/hed/src/dataset.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/research/cv/hed/src/dataset.py b/research/cv/hed/src/dataset.py
index 842d1438d..74233afd4 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
--
GitLab