VanLinLin commited on
Commit
e242910
·
1 Parent(s): 98ea233

update .pth name

Browse files
Files changed (1) hide show
  1. 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='best_model_densefusion.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')
 
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')