Skip to content

Commit 1de8bab

Browse files
delete CenterCrop for ic4
For ic4, image shape should be (299, 299), it don't need CenterCrop. Co-authored-by: xz-linghu <[email protected]>
1 parent d8bcf3f commit 1de8bab

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

examples/samsung/scripts/inception_v4.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,14 +28,12 @@ def get_dataset(dataset_path, data_size):
2828
from torchvision import datasets, transforms
2929

3030
image_shape = (299, 299)
31-
crop_size = 224
3231
shuffle = True
3332

3433
def get_data_loader():
3534
preprocess = transforms.Compose(
3635
[
3736
transforms.Resize(image_shape),
38-
transforms.CenterCrop(crop_size),
3937
transforms.ToTensor(),
4038
transforms.Normalize(
4139
mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]

0 commit comments

Comments
 (0)