Commit Graph

38 Commits (3a4e390c77a8e11e2b30eabd2a1113d501ff08fc)

Author SHA1 Message Date
KaiyangZhou 3a4e390c77 update '==>' to '=>' 2018-11-13 16:07:41 +00:00
KaiyangZhou 8c53007138 return (imgs, pids, camids, img_paths) for image dataloaders 2018-11-10 11:54:06 +00:00
KaiyangZhou 4a2b98c3b7 update --eval-step to --eval-freq 2018-11-09 21:58:58 +00:00
KaiyangZhou 9d1f313f4f add args.fixbase 2018-11-09 00:11:11 +00:00
KaiyangZhou b85e1b59d6 finalize fixbase functions 2018-11-09 00:02:46 +00:00
KaiyangZhou 57682b768e add ranklogger; update variable names 2018-11-08 21:41:32 +00:00
KaiyangZhou 893152a335 adapt to changes in data_manager 2018-11-07 21:46:39 +00:00
KaiyangZhou 3f9e20a55e change access to dataloaders in dm by class method 2018-11-07 20:48:21 +00:00
KaiyangZhou 506369f9ad correct test_batch to test_batch_size 2018-11-07 17:13:39 +00:00
KaiyangZhou e6464f3443 update args for dataset/optimizer 2018-11-07 17:09:23 +00:00
KaiyangZhou 2ecd8e5704 wrap argument_parser in args.py 2018-11-07 15:36:49 +00:00
KaiyangZhou 7587dfe7ec fix typo: criterion -> criterion_xent 2018-11-05 21:38:58 +00:00
KaiyangZhou a96dd6b916 support multi-dataset training 2018-11-05 21:36:12 +00:00
KaiyangZhou c388c24880 wrap transforms into build_transforms() 2018-11-01 22:29:19 +00:00
KaiyangZhou a57558738e generalize CrossEntropyLableSmooth to CrossEntropyLoss 2018-10-29 23:00:17 +00:00
KaiyangZhou 478da99655 update isinstance(outputs, (tuple, list)) 2018-10-25 17:45:00 +01:00
KaiyangZhou 0374e33c7c add 'global' keyword to args, best_rank1 2018-10-23 11:37:03 +01:00
KaiyangZhou 1fa306e90d mv --vis-ranked-res to --visualize-ranks 2018-10-06 20:51:20 +01:00
KaiyangZhou cfd0fea779 make 'best_rank1' a global variable; best_epoch = args.start_epoch; 2018-09-29 22:51:56 +01:00
KaiyangZhou 9356c70de3 add --use-avai-gpus 2018-09-19 21:15:17 +01:00
KaiyangZhou 16d0ba14f9 add argument '--label-smooth' (default criterion is nn.CrossEntropyLoss()) 2018-08-30 09:36:54 +01:00
KaiyangZhou 9aa1aa24c0 update sampler for htri so pids do not repeat in each epoch 2018-08-16 19:11:37 +01:00
KaiyangZhou 611ac80f0b mv deepreid/ to torchreid/ 2018-08-15 09:48:17 +01:00
KaiyangZhou 0c18858f92 move codes to deepreid/ 2018-08-14 22:38:55 +01:00
KaiyangZhou bd18a9489e add lambda-xent and lambda-htri to balance training 2018-08-14 17:32:06 +01:00
KaiyangZhou 28ecca9397 rm lmdb 2018-08-11 22:22:48 +01:00
KaiyangZhou fce7779381 add check_isfile 2018-08-07 17:14:44 +01:00
KaiyangZhou 68b8ea01d7 change start_epoch to args.start_epoch 2018-08-01 14:59:43 +01:00
KaiyangZhou dbbb68b955 add vis-ranked-res 2018-08-01 12:04:58 +01:00
KaiyangZhou d9359d99bd add load_weights 2018-07-10 11:56:30 +01:00
KaiyangZhou b88d36cd9c add function of fixbase 2018-07-06 11:03:38 +01:00
KaiyangZhou d0db61231a standardize code 2018-07-04 10:32:43 +01:00
KaiyangZhou 9dd1d48d3c increase precision from .3f to .4f for DataTime 2018-07-03 09:55:22 +01:00
KaiyangZhou c2dab79875 change computation of elapsed time 2018-07-02 16:39:52 +01:00
KaiyangZhou 7b04fce8a3 update code 2018-07-02 13:57:11 +01:00
KaiyangZhou 4b589353ff add support for reading lmdb dataset 2018-07-02 12:55:28 +01:00
KaiyangZhou 7d1c14f321 rm test code 2018-07-02 11:56:30 +01:00
KaiyangZhou 0de31316d3 update model & script 2018-07-02 10:17:14 +01:00