Skip to content
Snippets Groups Projects
Commit 0c417c8b authored by hexiangdong2019's avatar hexiangdong2019
Browse files

取消eval的preprocess

parent 0b7a7e4c
No related branches found
No related tags found
No related merge requests found
......@@ -19,6 +19,7 @@ from dataset import GetDatasetGenerator
from loss import SoftmaxCrossEntropyLoss
from learning_rates import exponential_lr, cosine_lr
context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False,
device_target='Ascend', device_id=7)
# net = UNetMedical(n_channels=3, n_classes=6)
......@@ -92,7 +93,8 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True):
resize_hw = []
for l in range(batch_size):
img_ = img_lst[l]
img_, resize_h, resize_w = pre_process(img_, crop_size)
resize_h, resize_w, _ = img_.shape
# img_, resize_h, resize_w = pre_process(img_, crop_size)
batch_img[l] = img_
resize_hw.append([resize_h, resize_w])
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment