set msls train/val cities default args

Author: oravus

README.md

@@ -42,9 +42,9 @@ or the Oxford dataset (set `--dataset oxford-pnv` for pointnetvlad-like data spl
 ```python
 python main.py --mode train --pooling seqnet --dataset oxford-v1.0 --seqL 5 --w 3 --outDims 4096 --expName "w3"
 ```
-or the MSLS dataset (must specify `--msls_trainCity` and `--msls_valCity`):
+or the MSLS dataset (specifying `--msls_trainCity` and `--msls_valCity` as default values):
 ```python
-python main.py --mode train --pooling seqnet --dataset msls --msls_trainCity melbourne --msls_valCity austin --seqL 5 --w 3 --outDims 4096 --expName "msls_w5"
+python main.py --mode train --pooling seqnet --dataset msls --msls_trainCity melbourne --msls_valCity austin --seqL 5 --w 3 --outDims 4096 --expName "msls_w3"
 ```
 
 To train transformed single descriptors through SeqNet:
@@ -57,9 +57,9 @@ On the Nordland dataset:
 ```python
 python main.py --mode test --pooling seqnet --dataset nordland-sf --seqL 5 --split test --resume ./data/runs/Jun03_15-22-44_l10_w5/ 
 ```
-On the MSLS dataset:
+On the MSLS dataset (can change `--msls_valCity` to `melbourne` or `austin` too):
 ```python
-python main.py --mode test --pooling seqnet --dataset msls --msls_trainCity melbourne --msls_valCity austin --seqL 5 --split test --resume ./data/runs/<modelName>/
+python main.py --mode test --pooling seqnet --dataset msls --msls_valCity amman --seqL 5 --split test --resume ./data/runs/<modelName>/
 ```
 
 The above will reproduce results for SeqNet (S5) as per [Supp. Table III on Page 10](https://arxiv.org/pdf/2102.11603.pdf).

main.py

@@ -57,8 +57,8 @@
 
 # dataset, model etc.
 parser.add_argument('--dataset', type=str, default='nordland-sw', help='Dataset to use', choices=['nordland-sw', 'nordland-sf', 'oxford-v1.0', 'oxford-pnv', 'msls'])
-parser.add_argument('--msls_trainCity', type=str, help='trainCityName')
-parser.add_argument('--msls_valCity', type=str, help='valCityName')
+parser.add_argument('--msls_trainCity', type=str, default='melbourne', help='trainCityName')
+parser.add_argument('--msls_valCity', type=str, default='austin', help='valCityName')
 parser.add_argument('--pooling', type=str, default='seqnet', help='type of pooling to use', choices=[ 'seqnet', 'smooth', 'delta', 'single','single+seqmatch', 's1+seqmatch'])
 parser.add_argument('--seqL', type=int, default=5, help='Sequence Length')
 parser.add_argument('--w', type=int, default=3, help='filter size for seqNet')