KaiyangZhou
|
d72b85a1b8
|
set output_featuremaps_only=False in osnet
|
2019-08-03 21:52:52 +01:00 |
|
KaiyangZhou
|
7e0249a084
|
add output_featuremaps_only to foward() in osnet
|
2019-08-03 18:25:34 +01:00 |
|
KaiyangZhou
|
27d4f2f093
|
add visactmap code; upgrade visrank to plot single figure
|
2019-08-03 17:35:06 +01:00 |
kaiyangzhou
|
60d27cf9e9
|
add osnet code
|
2019-07-03 13:37:25 +01:00 |
|
KaiyangZhou
|
554807ad36
|
update mobilenetv2_1dot0 => mobilenetv2_x1_0 and the same for x1_4
|
2019-07-03 11:33:10 +01:00 |
|
KaiyangZhou
|
752cb82682
|
change attribute name: last_linear -> classifier
|
2019-06-03 09:26:04 +01:00 |
|
KaiyangZhou
|
9963cff629
|
update imagenet weights path
|
2019-05-28 15:33:04 +01:00 |
|
kaiyangzhou
|
e3c326677a
|
add public keys to comments
|
2019-05-24 16:15:55 +01:00 |
|
kaiyangzhou
|
fd88366d22
|
rm old resnext
|
2019-05-24 15:35:54 +01:00 |
|
kaiyangzhou
|
97ee4b423b
|
merge resnet and resnext; add shufflenetv2
|
2019-05-24 15:35:09 +01:00 |
|
KaiyangZhou
|
28b4580a0d
|
remove print in internal wrapper
|
2019-04-23 11:35:22 +01:00 |
|
KaiyangZhou
|
8e5265099d
|
change weight path
|
2019-03-25 09:29:06 +00:00 |
|
KaiyangZhou
|
4ca182dbb8
|
add model public keys
|
2019-03-24 15:52:39 +00:00 |
|
KaiyangZhou
|
aa98ef5a7b
|
add docstrings to losses, metrics, models, utils, optim
|
2019-03-22 00:14:41 +00:00 |
|
KaiyangZhou
|
37f92b7cc3
|
create engine SDK
|
2019-03-19 17:26:08 +00:00 |
|
KaiyangZhou
|
03d0674dcb
|
comply woth PEP8
|
2019-03-15 23:17:38 +00:00 |
|
KaiyangZhou
|
850da2bbed
|
update path to pretrained weights of resnext
|
2019-03-12 16:46:49 +00:00 |
|
KaiyangZhou
|
7ff331f871
|
update mobilenetv2; add imagenet/reid weights
|
2019-03-05 10:56:39 +00:00 |
|
kaiyangzhou
|
859b03cf19
|
set pretrained=True instead of pretrained="imagenet"
|
2019-02-19 11:10:45 +00:00 |
|
KaiyangZhou
|
564f5a26ab
|
update print
|
2019-01-30 22:41:47 +00:00 |
|
KaiyangZhou
|
c05f96e31b
|
add link to shufflenet's imagenet weights
|
2018-11-26 22:07:20 +00:00 |
|
KaiyangZhou
|
47b883d0d6
|
bug fix: rm resnext101_32x4d
|
2018-11-26 22:06:33 +00:00 |
|
KaiyangZhou
|
affeaa847d
|
add resnext50_32x4d's imagenet weights
|
2018-11-23 10:31:38 +00:00 |
|
KaiyangZhou
|
d7755cfd16
|
add resnext50_32x4d_fc512
|
2018-11-22 12:41:22 +00:00 |
|
KaiyangZhou
|
0545358b6f
|
mv resnet50mid to reid-specific model list
|
2018-11-09 16:17:29 +00:00 |
|
KaiyangZhou
|
81d3cc3995
|
fixed division error: / -> //
|
2018-11-05 21:30:40 +00:00 |
|
KaiyangZhou
|
39df4855c1
|
add mlfn
|
2018-11-04 00:18:02 +00:00 |
|
KaiyangZhou
|
e302af017d
|
recover dropout in pcb
|
2018-11-01 17:28:15 +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
|
4bf76590df
|
change F.tanh/F.sigmoid to torch.tanh/torch.sigmoid
|
2018-09-22 17:45:34 +01:00 |