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')
|