Commit Graph

271 Commits (069356a6e4915fa01b7abb51a15b5dfd816db029)

Author SHA1 Message Date
KaiyangZhou 75279f0728 add evaluate_py 2018-11-10 21:25:40 +00:00
KaiyangZhou 41f1f6c6ac fix bug: add return 2018-11-10 21:16:31 +00:00
KaiyangZhou d885f11ce7 update cython code: add cuhk03-metric 2018-11-10 21:09:13 +00:00
KaiyangZhou 1d0851e8c4 fix bug: add self.num_instances to ImageDataManger 2018-11-10 11:59:01 +00:00
KaiyangZhou 8c53007138 return (imgs, pids, camids, img_paths) for image dataloaders 2018-11-10 11:54:06 +00:00
KaiyangZhou db157c1345 update help comments for data_source 2018-11-10 11:53:25 +00:00
KaiyangZhou 0545358b6f mv resnet50mid to reid-specific model list 2018-11-09 16:17:29 +00:00
KaiyangZhou 54b92ee617 check if isinstance(model, nn.DataParallel) 2018-11-08 23:54:26 +00:00
KaiyangZhou 205df55987 add fixbase functions (debug mode) 2018-11-08 23:04:50 +00:00
KaiyangZhou cac6d72a6d minor change: data_source (Dataset) -> data_source (list) 2018-11-08 22:34:24 +00:00
KaiyangZhou 9980b5378b add support for htri training 2018-11-08 22:02:53 +00:00
KaiyangZhou 01fe76cf27 sample -> sample_method 2018-11-08 21:40:44 +00:00
KaiyangZhou e17bbcddf3 source/target -> source_names/target_names 2018-11-08 21:40:18 +00:00
KaiyangZhou c32cf80a9f add RankLogger 2018-11-08 21:38:22 +00:00
KaiyangZhou fdc451c0e5 add linear decay 2018-11-07 22:02:23 +00:00
KaiyangZhou 2d32f5b28a add warnings 2018-11-07 21:55:04 +00:00
KaiyangZhou e3ae601ffc mv KeyError to ValueError 2018-11-07 21:54:46 +00:00
KaiyangZhou 7b47e645e9 add property 2018-11-07 21:46:23 +00:00
KaiyangZhou 1f1f01669b change dataset to (query, gallery) 2018-11-07 21:45:48 +00:00
KaiyangZhou 3f9e20a55e change access to dataloaders in dm by class method 2018-11-07 20:48:21 +00:00
KaiyangZhou e6464f3443 update args for dataset/optimizer 2018-11-07 17:09:23 +00:00
KaiyangZhou b81c13fd10 fix bug: query -> gallery 2018-11-07 15:54:10 +00:00
KaiyangZhou 2ecd8e5704 wrap argument_parser in args.py 2018-11-07 15:36:49 +00:00
KaiyangZhou a96dd6b916 support multi-dataset training 2018-11-05 21:36:12 +00:00
KaiyangZhou 81d3cc3995 fixed division error: / -> // 2018-11-05 21:30:40 +00:00
KaiyangZhou 9726b8bb96 set required=True for dataset arguments 2018-11-05 21:25:09 +00:00
KaiyangZhou 02bac86c71 add data_manager.py 2018-11-05 21:19:27 +00:00
KaiyangZhou e96ac0e3a4 add warning to non-cython evaluation 2018-11-05 19:24:20 +00:00
KaiyangZhou df0ae6d0e0 add num_cams; create base classes 2018-11-04 22:59:46 +00:00
KaiyangZhou 01d7594011 make camid start from 0 2018-11-04 21:27:08 +00:00
KaiyangZhou 39df4855c1 add mlfn 2018-11-04 00:18:02 +00:00
KaiyangZhou 733d967345 rm multi-crop 2018-11-02 11:39:19 +00:00
KaiyangZhou c388c24880 wrap transforms into build_transforms() 2018-11-01 22:29:19 +00:00
KaiyangZhou e302af017d recover dropout in pcb 2018-11-01 17:28:15 +00:00
KaiyangZhou a57558738e generalize CrossEntropyLableSmooth to CrossEntropyLoss 2018-10-29 23:00:17 +00:00
KaiyangZhou 752a4fe498 update pcb: commented dropout 2018-10-28 22:30:19 +00:00
KaiyangZhou 7588c8f581 update pcb: add support to various resnet archs 2018-10-28 11:44:09 +00:00
KaiyangZhou 12cdbf96b5 update pcb: mv conv5 after eval-mode in forward 2018-10-28 11:30:35 +00:00
KaiyangZhou bfba6389d6 add pcb 2018-10-28 11:16:42 +00:00
KaiyangZhou f77b9ada10 fill space after comma 2018-10-27 23:22:23 +01:00
KaiyangZhou bc942aaf09 add variable model_url 2018-10-27 22:50:22 +01:00
KaiyangZhou 3ddf9ce699 build new classifier; add squeezenet1_0_fc512 2018-10-27 22:49:53 +01:00
KaiyangZhou ee5bcee63f add delimiter for lightweight models 2018-10-27 17:59:10 +01:00
KaiyangZhou afca4b442e add delimiter to separate image-classification models and reid models 2018-10-27 17:49:11 +01:00
KaiyangZhou 4ce9024fcc add reference to all models 2018-10-27 17:46:48 +01:00
KaiyangZhou d187902ed4 add resnext 2018-10-27 16:52:05 +01:00
KaiyangZhou 2c2a9c0098 update model code 2018-10-27 14:42:28 +01:00
KaiyangZhou 62aad1324a add import 2018-10-27 14:12:51 +01:00
KaiyangZhou 0af4b88fba add reference 2018-10-27 14:09:21 +01:00
KaiyangZhou 345093a0ba change pretrained=True 2018-10-27 12:01:51 +01:00
KaiyangZhou 35235fe4bb update densenet 2018-10-27 11:57:48 +01:00
KaiyangZhou d076e41e78 add def featuremaps(self, x) 2018-10-27 10:49:27 +01:00
KaiyangZhou fffa92d55c add self.loss 2018-10-27 10:40:20 +01:00
KaiyangZhou b5c305fcce add **kwargs to class __init__() 2018-10-27 10:35:47 +01:00
KaiyangZhou ee5fb676bf add dropout_p; update senet 2018-10-26 23:54:13 +01:00
KaiyangZhou 710b9ac1a2 update model code 2018-10-26 23:05:40 +01:00
KaiyangZhou 1feb4644b9 update split generation in cuhk01 2018-10-10 16:10:26 +01:00
KaiyangZhou 70365320f5 correct split generation in viper 2018-10-08 16:44:26 +01:00
KaiyangZhou 3bd674f198 use imgname as dirname for each query 2018-10-06 22:21:11 +01:00
KaiyangZhou e6ce128a26 update print 2018-10-06 22:16:28 +01:00
KaiyangZhou 86eb3336c9 update print 2018-10-06 22:16:07 +01:00
KaiyangZhou b0aa35cb03 add image_name to query_dir_name 2018-10-06 22:09:44 +01:00
KaiyangZhou 1fa306e90d mv --vis-ranked-res to --visualize-ranks 2018-10-06 20:51:20 +01:00
KaiyangZhou 4bf76590df change F.tanh/F.sigmoid to torch.tanh/torch.sigmoid 2018-09-22 17:45:34 +01:00
KaiyangZhou 95ede066c7 add isinstance(src, list) check for img_paths in tracklets 2018-09-04 10:11:43 +01:00
KaiyangZhou 3bc884770f add sensereid dataset 2018-08-17 10:46:19 +01:00
KaiyangZhou 9d62ea19ac rm data_manager/base.py 2018-08-17 10:45:59 +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 b00140cd61 automate download 2018-08-15 14:29:44 +01:00
KaiyangZhou 64d06bb47c automate the download of dukemtmcreid 2018-08-15 13:58:18 +01:00
KaiyangZhou 611ac80f0b mv deepreid/ to torchreid/ 2018-08-15 09:48:17 +01:00