Skip to content

T
tongue-diagnosis
Commit 649c2e07 authored by ZhouZhiyang's avatar ZhouZhiyang
Browse files
Options

注释normalizer

parent 5895e5cb
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train_tongue_diagnosis.py
View file @ 649c2e07
...@@ -76,7 +76,7 @@ transform_test = transforms.Compose([ ...@@ -76,7 +76,7 @@ transform_test = transforms.Compose([
transforms.ToPILImage(), transforms.ToPILImage(),
transforms.Resize((298, 224)), transforms.Resize((298, 224)),
transforms.ToTensor(), transforms.ToTensor(),
normalizer # normalizer
]) ])
test_data_dir = args.test_data_dir test_data_dir = args.test_data_dir
tongue_dataset_test = Image_DataSet(processed_dir=test_data_dir, transform=transform_test) tongue_dataset_test = Image_DataSet(processed_dir=test_data_dir, transform=transform_test)
......
Markdown is supported
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