575 Commits

Author SHA1 Message Date
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
478da99655 update isinstance(outputs, (tuple, list)) 2018-10-25 17:45:00 +01:00
KaiyangZhou
602971e4d7 update requirements.txt (rm version specifications) 2018-10-23 18:43:20 +01:00
KaiyangZhou
0374e33c7c add 'global' keyword to args, best_rank1 2018-10-23 11:37:03 +01:00
KaiyangZhou
e9b578a720 update instructions for cuhk01 2018-10-10 16:12:31 +01:00
KaiyangZhou
1feb4644b9 update split generation in cuhk01 2018-10-10 16:10:26 +01:00
KaiyangZhou
11d7edc6cc update readme 2018-10-09 17:02:47 +01:00
KaiyangZhou
a80bfb6193 update readme 2018-10-09 10:18:36 +01:00
KaiyangZhou
472f6836f2 add RELATED_PROJECTS.md 2018-10-09 10:14:45 +01:00
KaiyangZhou
4b8e4328fa update readme 2018-10-09 10:14:27 +01:00
KaiyangZhou
3ad38996d7 add instructions on evaluation 2018-10-09 10:14:13 +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
32ae8f8e4c update torch>=0.4.0 2018-10-01 11:50:25 +01:00
KaiyangZhou
cfd0fea779 make 'best_rank1' a global variable; best_epoch = args.start_epoch; 2018-09-29 22:51:56 +01:00
KaiyangZhou
4bf76590df change F.tanh/F.sigmoid to torch.tanh/torch.sigmoid 2018-09-22 17:45:34 +01:00
KaiyangZhou
9356c70de3 add --use-avai-gpus 2018-09-19 21:15:17 +01:00
KaiyangZhou
95ede066c7 add isinstance(src, list) check for img_paths in tracklets 2018-09-04 10:11:43 +01:00
KaiyangZhou
7c62ab2ed8 fig bug: replace dataset.train in triplet sampler with new_train 2018-08-31 11:04:44 +01:00
KaiyangZhou
16d0ba14f9 add argument '--label-smooth' (default criterion is nn.CrossEntropyLoss()) 2018-08-30 09:36:54 +01:00
KaiyangZhou
0084a30c6a fix bug: add 'args.pool' to test() 2018-08-28 09:57:12 +01:00
KaiyangZhou
cc1e23049a add instructions for sensereid 2018-08-17 10:55:27 +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 v0.3.0 2018-08-15 14:29:44 +01:00
KaiyangZhou
753d7c7d64 update readme: automate dukemtmcvidreid 2018-08-15 14:29:24 +01:00