kaiyangzhou
cf553d367f
bug fix: pass video-specific args to dataset.__add__
2019-10-04 12:09:28 +01:00
kaiyangzhou
7a73dca4ac
bug fix: img_path->path
2019-09-13 23:23:13 +01:00
KaiyangZhou
0a60d2c6a7
version=1.0.2
2019-09-01 15:04:14 +01:00
KaiyangZhou
0ec14d9afc
add random patch data augmentation
2019-09-01 15:03:42 +01:00
KaiyangZhou
9131c537d5
[upgrade to 1.0.1] add cuhk02
2019-08-29 17:05:29 +01:00
KaiyangZhou
5507b38c86
update v0.9.4 to v1.0.0
2019-08-26 13:26:10 +01:00
KaiyangZhou
5365619606
add config
2019-08-26 10:34:31 +01:00
KaiyangZhou
046ff13c07
minor change
2019-08-23 22:09:35 +01:00
KaiyangZhou
d9cdb584e2
0.9.4
2019-08-23 22:05:09 +01:00
KaiyangZhou
f4451e5ea9
fix visrank bug for video reid
2019-08-23 22:04:36 +01:00
Abner Ayala-Acevedo
2383698d52
Fix Visual Rank Video Reid Error
...
Fix video reid error while making sure it still works for image reid.
When trying to save visual ranks for video reid. It crashes on statement `qdir = osp.join(save_dir, osp.basename(osp.splitext(qimg_path)[0]))` that is because qimg_path can be a tuple or list for video reid. Hence, osp.splittext gives error when input is not a str. To fix this I make sure that osp.splitext always gets pass an str regardless if we doing video or image reid.
Let me know if you have any questions.
2019-08-23 12:12:32 -07:00
KaiyangZhou
a23817173c
upgrade to 0.9.3
2019-08-22 23:32:23 +01:00
KaiyangZhou
a40ba19207
add cosine lr
2019-08-22 23:17:47 +01:00
KaiyangZhou
92f1ea215a
add tensorboard
2019-08-22 22:41:21 +01:00
KaiyangZhou
1558389f37
osnet.py: rename residual->identity; engine: add num_batches=len(dloader)
2019-08-21 23:02:05 +01:00
KaiyangZhou
a20f8a66bd
output_featuremaps_only -> return_featuremaps
2019-08-05 14:01:39 +01:00
KaiyangZhou
d5863e0e82
0.9.1
2019-08-03 23:22:44 +01:00
KaiyangZhou
1732d9e5da
0.9.0
2019-08-03 23:15:42 +01:00
KaiyangZhou
e4fb577693
update docs for visrank and visactmap
2019-08-03 23:12:37 +01:00
KaiyangZhou
84249ad412
[update visactmap] change channel order
2019-08-03 23:01:21 +01:00
KaiyangZhou
d021fafff1
set visrank_topk=10 as default
2019-08-03 22:59:10 +01:00
KaiyangZhou
28189cf1a6
[update visrank] use grid_img instead of plt.imshow
2019-08-03 22:30:22 +01:00
KaiyangZhou
d72b85a1b8
set output_featuremaps_only=False in osnet
2019-08-03 21:52:52 +01:00
KaiyangZhou
a08a85d8e4
[update visactmap] use cv2.imwrite instead of pyplot
2019-08-03 21:42:50 +01:00
KaiyangZhou
7e0249a084
add output_featuremaps_only to foward() in osnet
2019-08-03 18:25:34 +01:00
KaiyangZhou
3d64d9eaed
add default font size to visactmap
2019-08-03 18:07:51 +01:00
KaiyangZhou
a6a5d39f68
print progress in visrank
2019-08-03 17:49:19 +01:00
KaiyangZhou
27d4f2f093
add visactmap code; upgrade visrank to plot single figure
2019-08-03 17:35:06 +01:00
kaiyangzhou
d398e66b37
allow empty transforms
2019-07-17 12:45:58 +01:00
KaiyangZhou
1f891c1b36
[upgrade] 0.8.0=>0.8.1
2019-07-08 15:47:56 +01:00
KaiyangZhou
9e418c0f5a
print resize(height, width)
2019-07-08 15:41:38 +01:00
KaiyangZhou
d20b63c1a0
resize image to target size
2019-07-05 09:05:03 +01:00
rick-hao
87fd094a2c
Update datamanager.py
2019-07-04 16:39:11 +10:00
KaiyangZhou
65a8f44d80
[datamanager] add comment
2019-07-03 15:54:20 +01:00
KaiyangZhou
6829805723
[version] 0.7.8 => 0.8.0
2019-07-03 13:40:23 +01:00
kaiyangzhou
60d27cf9e9
add osnet code
2019-07-03 13:37:25 +01:00
kaiyangzhou
709018d948
add argument 'transforms' to datamanager
2019-07-03 12:05:55 +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
fb4d585127
(epoch+1)>=start_eval
2019-06-13 23:11:07 +01:00
KaiyangZhou
752cb82682
change attribute name: last_linear -> classifier
2019-06-03 09:26:04 +01:00
KaiyangZhou
86ddbd97aa
update comment
2019-05-30 14:25:55 +01:00
KaiyangZhou
250cd30ae2
0.7.8
2019-05-28 23:07:37 +01:00
KaiyangZhou
9963cff629
update imagenet weights path
2019-05-28 15:33:04 +01:00
KaiyangZhou
31068ca233
add global keyword
2019-05-24 17:13:30 +01:00
KaiyangZhou
5ae81344a0
0.7.7
2019-05-24 16:30:11 +01:00
kaiyangzhou
e3c326677a
add public keys to comments
2019-05-24 16:15:55 +01:00
kaiyangzhou
f57d88decf
move fixbase into train
2019-05-24 15:36:10 +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
ac89f94129
add collect_env_info
2019-05-24 15:34:27 +01:00
KaiyangZhou
4a36d27ebc
update comments
2019-05-22 22:22:55 +01:00
KaiyangZhou
8ec474daa7
add input_size to print
2019-05-22 21:46:48 +01:00
KaiyangZhou
575b456a92
0.7.6
2019-05-22 16:16:35 +01:00
KaiyangZhou
a8a51b3a54
update docstring
2019-05-22 16:15:42 +01:00
KaiyangZhou
26cf6be779
import compute_model_complexity
2019-05-22 16:14:47 +01:00
KaiyangZhou
e0a23bab07
add model_complexity.py
2019-05-22 16:14:04 +01:00
KaiyangZhou
b8ab3ad482
0.7.5
2019-05-09 23:49:18 +01:00
KaiyangZhou
79abbacb1c
add print msg
2019-05-09 23:21:04 +01:00
KaiyangZhou
3e10dc60dc
add --normalize-feature to args parser
2019-05-09 23:11:10 +01:00
KaiyangZhou
5839a026c3
add person reranking
2019-05-09 22:47:55 +01:00
KaiyangZhou
6342fb3055
check if fpath exists
2019-05-09 22:47:09 +01:00
KaiyangZhou
024913f73c
update readme
2019-05-06 10:48:03 +01:00
KaiyangZhou
6fd0ab364f
raise error if batch_size is smaller than num_instances
2019-05-06 10:47:48 +01:00
KaiyangZhou
d0af8eb891
0.7.4
2019-04-27 23:31:05 +01:00
KaiyangZhou
bb6dc46e21
bug fix in combineall
2019-04-27 23:30:32 +01:00
KaiyangZhou
a52d1289f8
add attribute _junk_pids to dataset class
2019-04-27 23:02:54 +01:00
KaiyangZhou
dcd8da565a
bug fix in lr_scheduler
2019-04-26 15:45:23 +01:00
KaiyangZhou
28b4580a0d
remove print in internal wrapper
2019-04-23 11:35:22 +01:00
KaiyangZhou
dee01229da
remove division by 1e+06 when counting model size
2019-04-23 11:34:52 +01:00
KaiyangZhou
4b93121496
minor
2019-04-19 11:52:02 +01:00
KaiyangZhou
1decc2473a
0.7.2->0.7.3
2019-04-18 12:10:54 +01:00
KaiyangZhou
1ea7787b6e
typo fix
2019-04-17 09:13:04 +01:00
KaiyangZhou
4a033659b0
do not decay lr while fixbase=True
2019-03-29 11:40:23 +00:00
KaiyangZhou
4cf18f0830
print domain info while doing evaluation
2019-03-27 22:30:08 +00:00
KaiyangZhou
1e9d466f42
move scheduler.step() to train()
2019-03-27 13:01:39 +00:00
KaiyangZhou
8c1c0084b8
0.7.1 -> 0.7.2
2019-03-25 20:08:31 +00:00
kaiyangzhou
eaba4f5cfa
rm prid450s
2019-03-25 20:03:43 +00:00
kaiyangzhou
5cc8043c4a
verbose=False after __add__
2019-03-25 20:03:21 +00:00
KaiyangZhou
866744914b
0.7.0 -> 0.7.1
2019-03-25 18:49:20 +00:00
KaiyangZhou
235348f672
bug fix in __add__
2019-03-25 18:43:19 +00:00
KaiyangZhou
1e7bbc3668
add \n
2019-03-25 16:27:39 +00:00
KaiyangZhou
8e5265099d
change weight path
2019-03-25 09:29:06 +00:00
KaiyangZhou
e5b8c464ca
update docstrings
2019-03-24 23:33:54 +00:00
KaiyangZhou
4ca182dbb8
add model public keys
2019-03-24 15:52:39 +00:00
KaiyangZhou
a0e54ede32
add scripts
2019-03-24 11:06:12 +00:00
KaiyangZhou
d0fe37c558
finalize docs
2019-03-23 23:09:39 +00:00
KaiyangZhou
aa98ef5a7b
add docstrings to losses, metrics, models, utils, optim
2019-03-22 00:14:41 +00:00
KaiyangZhou
cbbde86eef
add docs to torchreid.data
2019-03-21 17:28:14 +00:00
KaiyangZhou
75dd835e42
change batch_size in video
2019-03-21 14:12:50 +00:00
KaiyangZhou
77c47b4e53
add data module
2019-03-21 13:00:07 +00:00
KaiyangZhou
46a0d5c27d
update dataset and docs
2019-03-21 12:53:21 +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
983c9e51e5
add --combineall
2019-03-15 17:45:47 +00:00
KaiyangZhou
402e054110
rearrange order of functions
2019-03-15 16:20:47 +00:00
KaiyangZhou
3cbeab42c2
add init_attributes to BaseDataset
2019-03-15 16:11:25 +00:00
KaiyangZhou
e2b1001165
polish code
2019-03-15 14:49:18 +00:00
KaiyangZhou
850da2bbed
update path to pretrained weights of resnext
2019-03-12 16:46:49 +00:00
KaiyangZhou
6f3b057c1b
add space between class and __init__ (trivial)
2019-03-09 12:53:23 +00:00
KaiyangZhou
5cd0776eb5
rm center loss & ring loss
2019-03-09 12:45:33 +00:00