From 68ee506a98206393a0438c728955debd405c405f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lojze=20=C5=BDust?= Date: Sat, 23 Jul 2022 18:17:47 +0000 Subject: [PATCH] Configs + training README --- README.md | 12 +- configs/all_list_hist.txt | 1324 ++++++++++++++++++++++++++++++++++ configs/mastr1325_train.yaml | 5 + configs/mastr1325_val.yaml | 5 + configs/val_list_hist.txt | 26 + 5 files changed, 1366 insertions(+), 6 deletions(-) create mode 100644 configs/all_list_hist.txt create mode 100644 configs/mastr1325_train.yaml create mode 100644 configs/mastr1325_val.yaml create mode 100644 configs/val_list_hist.txt diff --git a/README.md b/README.md index acd072c..8fc2d56 100644 --- a/README.md +++ b/README.md @@ -102,7 +102,7 @@ Currently available pretrained model weights. All models are evaluated on the MO To train your own models, use the `train.py` script. For example, to reproduce the results of our experiments use the following steps: -1. Download and prepare the [MaSTr1325 dataset](https://box.vicos.si/borja/viamaro/index.html#mastr1325) (images and GT masks). Also download the context frames for the MaSTr1325 images [here](TODO). +1. Download and prepare the [MaSTr1325 dataset](https://box.vicos.si/borja/viamaro/index.html#mastr1325) (images and GT masks). Also download the context frames for the MaSTr1325 images [here](#data). 2. Edit the dataset configuration files (`configs/mastr_1325_train.yaml`, `configs/mastr1325_extra` and `configs/mastr1325_val.yaml`) so that they correctly point to the dataset directories. 3. Use the `train.py` to train the network. @@ -111,10 +111,10 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3 # GPUs to use python train.py \ --train_config configs/mastr1325_train.yaml \ --val_config configs/mastr1325_val.yaml \ ---model_name my_wasr \ --validation \ ---batch_size 4 \ ---epochs 50 +--model_name my_wasr \ +--batch_size 2 \ +--epochs 100 ``` @@ -130,9 +130,9 @@ tensorboard --logdir output/logs/model_name ## Data +We extend the MaSTr1325 dataset by providing the context frames (5 preceding frames). We also extend the dataset with additional hard examples to form MaSTr1478. - MaSTr1325 context frames: TODO -- MaSTr1478 extension data: TODO -- MODS context frames: TODO +- MaSTr1478 extension data: Coming soon ## Citation diff --git a/configs/all_list_hist.txt b/configs/all_list_hist.txt new file mode 100644 index 0000000..c3e1512 --- /dev/null +++ b/configs/all_list_hist.txt @@ -0,0 +1,1324 @@ +0001 +0002 +0003 +0004 +0005 +0006 +0007 +0008 +0009 +0010 +0011 +0012 +0013 +0014 +0015 +0016 +0017 +0018 +0019 +0020 +0021 +0022 +0023 +0024 +0025 +0026 +0027 +0028 +0029 +0030 +0031 +0032 +0033 +0034 +0035 +0036 +0037 +0038 +0039 +0040 +0041 +0042 +0043 +0044 +0045 +0046 +0047 +0048 +0049 +0050 +0051 +0052 +0053 +0054 +0055 +0056 +0057 +0058 +0059 +0060 +0061 +0062 +0063 +0064 +0065 +0066 +0067 +0068 +0069 +0070 +0071 +0072 +0073 +0074 +0075 +0076 +0077 +0078 +0079 +0080 +0081 +0082 +0083 +0084 +0085 +0086 +0087 +0088 +0089 +0090 +0091 +0092 +0093 +0094 +0095 +0096 +0097 +0098 +0099 +0100 +0101 +0102 +0103 +0104 +0105 +0106 +0107 +0108 +0109 +0110 +0111 +0112 +0113 +0114 +0115 +0116 +0117 +0118 +0119 +0120 +0121 +0122 +0123 +0124 +0125 +0126 +0127 +0128 +0129 +0130 +0131 +0132 +0133 +0134 +0135 +0136 +0137 +0138 +0139 +0140 +0141 +0142 +0143 +0150 +0151 +0152 +0156 +0157 +0158 +0159 +0164 +0165 +0166 +0167 +0168 +0169 +0170 +0171 +0172 +0173 +0174 +0175 +0176 +0177 +0178 +0179 +0180 +0181 +0182 +0183 +0184 +0185 +0186 +0187 +0188 +0189 +0190 +0191 +0192 +0193 +0194 +0195 +0196 +0197 +0198 +0199 +0200 +0201 +0202 +0203 +0204 +0205 +0206 +0207 +0208 +0209 +0210 +0211 +0212 +0213 +0214 +0215 +0216 +0217 +0218 +0219 +0220 +0221 +0222 +0223 +0224 +0225 +0226 +0227 +0228 +0229 +0230 +0231 +0232 +0233 +0234 +0235 +0236 +0237 +0238 +0239 +0240 +0241 +0242 +0243 +0244 +0245 +0246 +0247 +0248 +0249 +0250 +0251 +0252 +0253 +0254 +0255 +0256 +0257 +0258 +0259 +0260 +0261 +0262 +0263 +0264 +0265 +0266 +0267 +0268 +0269 +0270 +0271 +0272 +0273 +0274 +0275 +0276 +0277 +0278 +0279 +0280 +0281 +0282 +0283 +0284 +0285 +0286 +0287 +0288 +0289 +0290 +0291 +0292 +0293 +0294 +0295 +0296 +0297 +0298 +0299 +0300 +0301 +0302 +0303 +0304 +0305 +0306 +0307 +0308 +0309 +0310 +0311 +0312 +0313 +0314 +0315 +0316 +0317 +0318 +0319 +0320 +0321 +0322 +0323 +0324 +0325 +0326 +0327 +0328 +0329 +0330 +0331 +0332 +0333 +0334 +0335 +0336 +0337 +0338 +0339 +0340 +0341 +0342 +0343 +0344 +0345 +0346 +0347 +0348 +0349 +0350 +0351 +0352 +0353 +0354 +0355 +0356 +0357 +0358 +0359 +0360 +0361 +0362 +0363 +0364 +0365 +0366 +0367 +0368 +0369 +0370 +0371 +0372 +0373 +0374 +0375 +0376 +0377 +0378 +0379 +0380 +0381 +0382 +0383 +0384 +0385 +0386 +0387 +0388 +0389 +0390 +0391 +0392 +0393 +0394 +0395 +0396 +0397 +0398 +0399 +0400 +0401 +0402 +0403 +0404 +0405 +0406 +0407 +0408 +0409 +0410 +0411 +0412 +0413 +0414 +0415 +0416 +0417 +0418 +0419 +0420 +0421 +0422 +0423 +0424 +0425 +0426 +0427 +0428 +0429 +0430 +0431 +0432 +0433 +0434 +0435 +0436 +0437 +0438 +0439 +0440 +0441 +0442 +0443 +0444 +0445 +0446 +0447 +0448 +0449 +0450 +0451 +0452 +0453 +0454 +0455 +0456 +0457 +0458 +0459 +0460 +0461 +0462 +0463 +0464 +0465 +0466 +0467 +0468 +0469 +0470 +0471 +0472 +0473 +0474 +0475 +0476 +0477 +0478 +0479 +0480 +0481 +0482 +0483 +0484 +0485 +0486 +0487 +0488 +0489 +0490 +0491 +0492 +0493 +0494 +0495 +0496 +0497 +0498 +0499 +0500 +0501 +0502 +0503 +0504 +0505 +0506 +0507 +0508 +0509 +0510 +0511 +0512 +0513 +0514 +0515 +0516 +0517 +0518 +0519 +0520 +0521 +0522 +0523 +0524 +0525 +0526 +0527 +0528 +0529 +0530 +0531 +0532 +0533 +0534 +0535 +0536 +0537 +0538 +0539 +0540 +0541 +0542 +0543 +0544 +0545 +0546 +0547 +0548 +0549 +0550 +0551 +0552 +0553 +0554 +0555 +0556 +0557 +0558 +0559 +0560 +0561 +0562 +0563 +0564 +0565 +0566 +0567 +0568 +0569 +0570 +0571 +0572 +0573 +0574 +0575 +0576 +0577 +0578 +0579 +0580 +0581 +0582 +0583 +0584 +0585 +0586 +0587 +0588 +0589 +0590 +0591 +0592 +0593 +0594 +0595 +0596 +0597 +0598 +0599 +0600 +0601 +0602 +0603 +0604 +0605 +0606 +0607 +0608 +0609 +0610 +0611 +0612 +0613 +0614 +0615 +0616 +0617 +0618 +0619 +0620 +0621 +0622 +0623 +0624 +0625 +0626 +0627 +0628 +0629 +0630 +0631 +0632 +0633 +0634 +0635 +0636 +0637 +0638 +0639 +0640 +0641 +0642 +0643 +0644 +0645 +0646 +0647 +0648 +0649 +0650 +0651 +0652 +0653 +0654 +0655 +0656 +0657 +0658 +0660 +0661 +0663 +0664 +0665 +0666 +0667 +0668 +0669 +0670 +0671 +0672 +0673 +0674 +0675 +0676 +0677 +0678 +0679 +0680 +0681 +0682 +0683 +0684 +0686 +0687 +0688 +0689 +0690 +0691 +0692 +0693 +0694 +0695 +0696 +0697 +0698 +0699 +0700 +0701 +0702 +0703 +0704 +0705 +0706 +0707 +0708 +0709 +0710 +0711 +0712 +0713 +0714 +0715 +0716 +0717 +0718 +0719 +0720 +0721 +0722 +0723 +0724 +0725 +0726 +0727 +0728 +0729 +0730 +0731 +0732 +0733 +0734 +0735 +0736 +0737 +0738 +0739 +0740 +0741 +0742 +0743 +0744 +0745 +0746 +0747 +0748 +0749 +0750 +0751 +0752 +0753 +0754 +0755 +0756 +0757 +0758 +0759 +0760 +0761 +0762 +0763 +0764 +0765 +0766 +0767 +0768 +0769 +0770 +0771 +0772 +0773 +0774 +0775 +0776 +0777 +0778 +0779 +0780 +0781 +0782 +0783 +0784 +0785 +0786 +0787 +0788 +0789 +0790 +0791 +0792 +0793 +0794 +0795 +0796 +0797 +0798 +0799 +0800 +0801 +0802 +0803 +0804 +0805 +0806 +0807 +0808 +0809 +0810 +0811 +0812 +0813 +0814 +0815 +0816 +0817 +0818 +0819 +0820 +0821 +0822 +0823 +0824 +0825 +0826 +0827 +0828 +0829 +0830 +0831 +0832 +0833 +0834 +0835 +0836 +0837 +0838 +0839 +0840 +0841 +0842 +0843 +0844 +0845 +0846 +0847 +0848 +0849 +0850 +0851 +0852 +0853 +0854 +0855 +0856 +0857 +0858 +0859 +0860 +0861 +0862 +0863 +0864 +0865 +0866 +0867 +0868 +0869 +0870 +0871 +0872 +0873 +0874 +0875 +0876 +0877 +0878 +0879 +0880 +0881 +0882 +0883 +0884 +0885 +0886 +0887 +0888 +0889 +0890 +0891 +0892 +0893 +0894 +0895 +0896 +0897 +0898 +0899 +0900 +0901 +0902 +0903 +0904 +0905 +0906 +0907 +0908 +0909 +0910 +0911 +0912 +0913 +0914 +0915 +0916 +0917 +0918 +0919 +0920 +0921 +0922 +0923 +0924 +0925 +0926 +0927 +0928 +0929 +0930 +0931 +0932 +0933 +0934 +0935 +0936 +0937 +0938 +0939 +0940 +0941 +0942 +0943 +0944 +0945 +0946 +0947 +0948 +0949 +0950 +0951 +0952 +0953 +0954 +0955 +0956 +0957 +0958 +0959 +0960 +0961 +0962 +0963 +0964 +0965 +0966 +0967 +0968 +0969 +0970 +0971 +0972 +0973 +0974 +0975 +0976 +0977 +0978 +0979 +0980 +0981 +0982 +0983 +0984 +0985 +0986 +0987 +0988 +0989 +0990 +0991 +0992 +0993 +0994 +0995 +0996 +0997 +0998 +0999 +1000 +1001 +1002 +1003 +1004 +1005 +1006 +1007 +1008 +1009 +1010 +1011 +1012 +1013 +1014 +1015 +1016 +1017 +1018 +1019 +1020 +1021 +1022 +1023 +1024 +1025 +1026 +1027 +1028 +1029 +1030 +1031 +1032 +1033 +1034 +1035 +1036 +1037 +1038 +1039 +1040 +1041 +1042 +1043 +1044 +1045 +old_00000090L +old_00000330L +old_00000400L +old_00001001L +old_00001150L +old_00001200L +old_00001235L +old_00001300L +old_00001450L +old_00001500L +old_00002355L +old_00005600L +old_0001 +old_00010605L +old_00010690L +old_00011975L +old_0002 +old_0003 +old_0004 +old_0005 +old_0006 +old_0007 +old_0008 +old_0009 +old_0010 +old_0011 +old_0012 +old_0013 +old_0014 +old_0015 +old_0016 +old_0017 +old_0018 +old_0019 +old_0020 +old_0021 +old_0022 +old_0023 +old_0024 +old_0025 +old_0026 +old_0027 +old_0028 +old_0029 +old_0030 +old_0031 +old_0032 +old_0033 +old_0034 +old_0035 +old_0036 +old_0037 +old_0038 +old_0039 +old_0040 +old_0041 +old_0042 +old_0043 +old_0044 +old_0045 +old_0046 +old_0047 +old_0048 +old_0049 +old_0050 +old_0051 +old_0052 +old_0053 +old_0054 +old_0055 +old_0056 +old_0057 +old_0058 +old_0059 +old_0060 +old_0061 +old_0062 +old_0063 +old_0064 +old_0065 +old_0066 +old_0067 +old_0068 +old_0069 +old_0070 +old_0071 +old_0072 +old_0073 +old_0074 +old_0075 +old_0076 +old_0077 +old_0078 +old_0079 +old_0080 +old_0081 +old_0082 +old_0083 +old_0084 +old_0085 +old_0086 +old_0087 +old_0088 +old_0089 +old_0090 +old_0091 +old_0092 +old_0093 +old_0094 +old_0095 +old_0096 +old_0097 +old_0098 +old_0099 +old_0100 +old_0101 +old_0102 +old_0103 +old_0104 +old_0105 +old_0106 +old_0107 +old_0108 +old_0109 +old_0110 +old_0111 +old_0112 +old_0113 +old_0114 +old_0115 +old_0116 +old_0117 +old_0118 +old_0119 +old_0120 +old_0121 +old_0122 +old_0123 +old_0124 +old_0125 +old_0126 +old_0127 +old_0128 +old_0129 +old_0130 +old_0131 +old_0132 +old_0133 +old_0134 +old_0135 +old_0136 +old_0137 +old_0138 +old_0139 +old_0140 +old_0141 +old_0142 +old_0143 +old_0144 +old_0145 +old_0146 +old_0147 +old_0148 +old_0149 +old_0150 +old_0151 +old_0152 +old_0153 +old_0154 +old_0155 +old_0156 +old_0157 +old_0158 +old_0159 +old_0160 +old_0161 +old_0162 +old_0163 +old_0164 +old_0165 +old_0166 +old_0167 +old_0168 +old_0169 +old_0170 +old_0171 +old_0172 +old_0173 +old_0174 +old_0175 +old_0176 +old_0177 +old_0178 +old_0179 +old_0180 +old_0181 +old_0182 +old_0183 +old_0184 +old_0185 +old_0186 +old_0187 +old_0188 +old_0189 +old_0190 +old_0191 +old_0192 +old_0193 +old_0194 +old_0195 +old_0196 +old_0197 +old_0198 +old_0199 +old_0200 +old_0201 +old_0202 +old_0203 +old_0204 +old_0205 +old_0206 +old_0207 +old_0208 +old_0209 +old_0210 +old_0211 +old_0212 +old_0213 +old_0214 +old_0215 +old_0216 +old_0217 +old_0218 +old_0219 +old_0220 +old_0221 +old_0222 +old_0223 +old_0224 +old_0225 +old_0226 +old_0227 +old_0228 +old_0229 +old_0230 +old_0231 +old_0232 +old_0233 +old_0234 +old_0235 +old_0236 +old_0237 +old_0238 +old_0239 +old_0240 +old_0241 +old_0242 +old_0243 +old_0244 +old_0245 +old_0246 +old_0390 +old_0440 +old_0455 +old_0465 +old_0466 +old_0467 +old_0468 +old_0469 +old_0471 +old_0472 +old_0473 +old_0474 +old_0475 +old_0485 +old_0490 +old_0493 +old_0495 +old_0496 +old_0497 +old_0498 +old_0499 +old_0500 +old_0510 +old_0516 +old_0517 +old_0518 +old_0519 +old_0520 +old_0521 +old_0524 +old_0525 +old_0526 +old_0535 +old_0540 diff --git a/configs/mastr1325_train.yaml b/configs/mastr1325_train.yaml new file mode 100644 index 0000000..bdd8670 --- /dev/null +++ b/configs/mastr1325_train.yaml @@ -0,0 +1,5 @@ +image_dir: images +image_hist_dir: images_hist +image_hist_len: 5 +image_list: all_list_hist.txt +mask_dir: masks diff --git a/configs/mastr1325_val.yaml b/configs/mastr1325_val.yaml new file mode 100644 index 0000000..5b19ec3 --- /dev/null +++ b/configs/mastr1325_val.yaml @@ -0,0 +1,5 @@ +image_dir: images +image_hist_dir: images_hist +image_hist_len: 5 +image_list: val_list_hist.txt +mask_dir: masks diff --git a/configs/val_list_hist.txt b/configs/val_list_hist.txt new file mode 100644 index 0000000..2040df7 --- /dev/null +++ b/configs/val_list_hist.txt @@ -0,0 +1,26 @@ +0696 +0774 +1017 +old_0001 +0429 +0444 +0281 +0689 +0082 +0902 +0583 +old_0241 +0306 +0665 +0969 +0203 +0964 +0260 +0441 +0910 +0643 +0028 +0129 +0676 +0743 +0217