update .pth name
Browse files- test_shadow.py +1 -1
test_shadow.py
CHANGED
|
@@ -19,7 +19,7 @@ parser.add_argument('--input_dir', default='test_dir',
|
|
| 19 |
type=str, help='Directory of validation images')
|
| 20 |
parser.add_argument('--result_dir', default='./output_dir',
|
| 21 |
type=str, help='Directory for results')
|
| 22 |
-
parser.add_argument('--weights', default='
|
| 23 |
,type=str, help='Path to weights')
|
| 24 |
parser.add_argument('--arch', type=str, default='DenseSR', help='archtechture')
|
| 25 |
parser.add_argument('--batch_size', default=1, type=int, help='Batch size for dataloader')
|
|
|
|
| 19 |
type=str, help='Directory of validation images')
|
| 20 |
parser.add_argument('--result_dir', default='./output_dir',
|
| 21 |
type=str, help='Directory for results')
|
| 22 |
+
parser.add_argument('--weights', default='best_WSRD.pth'
|
| 23 |
,type=str, help='Path to weights')
|
| 24 |
parser.add_argument('--arch', type=str, default='DenseSR', help='archtechture')
|
| 25 |
parser.add_argument('--batch_size', default=1, type=int, help='Batch size for dataloader')
|