Change resize mechanism
Browse files
app.py
CHANGED
|
@@ -355,7 +355,9 @@ classes = [
|
|
| 355 |
|
| 356 |
transform = transforms.Compose([
|
| 357 |
|
| 358 |
-
transforms.Resize(
|
|
|
|
|
|
|
| 359 |
|
| 360 |
transforms.ToTensor(),
|
| 361 |
|
|
|
|
| 355 |
|
| 356 |
transform = transforms.Compose([
|
| 357 |
|
| 358 |
+
transforms.Resize(112),
|
| 359 |
+
|
| 360 |
+
transforms.CenterCrop(96),
|
| 361 |
|
| 362 |
transforms.ToTensor(),
|
| 363 |
|