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
|
b81c13fd10
|
fix bug: query -> gallery
|
2018-11-07 15:54:10 +00:00 |
|
KaiyangZhou
|
1b255992bb
|
update dataset to dm
|
2018-11-07 15:40:02 +00:00 |
|
KaiyangZhou
|
2ecd8e5704
|
wrap argument_parser in args.py
|
2018-11-07 15:36:49 +00:00 |
|
KaiyangZhou
|
bc13a9fdc1
|
change BENCHMARK.md to MODEL_ZOO.md
|
2018-11-06 15:21:48 +00:00 |
|
KaiyangZhou
|
1cefe7b5f1
|
update data_manager to datasets
|
2018-11-06 13:52:57 +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
|
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
|
ee44188b74
|
support multi-dataset training
|
2018-11-05 21:18:56 +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
|
c12b29bbfb
|
update requirements: torch>=0.4.1
|
2018-11-04 19:06:17 +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
|
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 |