93b8c9f3db
v1.3.6: added University-1652
2021-02-14 11:25:24 +08:00
9b73eb41d5
Add the GPU-Re-Ranking
2021-01-15 15:29:44 +08:00
626e6c3bb8
minor change in feature_extractor.py
2020-10-19 12:06:30 +01:00
fefa31dc3f
1.3.3: fixed bug in visrank (forgot unpacking dsetid)
2020-08-18 22:17:35 +01:00
5ce43185b1
linting code
2020-05-05 15:58:00 +01:00
8e5fc471a6
1.2.1: add simple API for feature extraction
2020-05-04 12:35:45 +01:00
36e22e8ce1
make engine more model-agnostic
2020-04-16 12:46:15 +01:00
469a038d01
add scheduler to resume_from_checkpoint
2019-12-10 10:44:45 +00:00
08bb2cc5eb
linting and formatting code
2019-12-02 14:34:59 +00:00
6b8fe56638
linting and formatting code
2019-12-01 02:35:44 +00:00
a136384a6a
rm unused imports/variables; fixed minor bugs
2019-10-23 17:15:37 +01:00
7a73dca4ac
bug fix: img_path->path
2019-09-13 23:23:13 +01:00
5365619606
add config
2019-08-26 10:34:31 +01:00
046ff13c07
minor change
2019-08-23 22:09:35 +01:00
f4451e5ea9
fix visrank bug for video reid
2019-08-23 22:04:36 +01:00
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
d021fafff1
set visrank_topk=10 as default
2019-08-03 22:59:10 +01:00
28189cf1a6
[update visrank] use grid_img instead of plt.imshow
2019-08-03 22:30:22 +01:00
a6a5d39f68
print progress in visrank
2019-08-03 17:49:19 +01:00
27d4f2f093
add visactmap code; upgrade visrank to plot single figure
2019-08-03 17:35:06 +01:00
ac89f94129
add collect_env_info
2019-05-24 15:34:27 +01:00
4a36d27ebc
update comments
2019-05-22 22:22:55 +01:00
8ec474daa7
add input_size to print
2019-05-22 21:46:48 +01:00
a8a51b3a54
update docstring
2019-05-22 16:15:42 +01:00
26cf6be779
import compute_model_complexity
2019-05-22 16:14:47 +01:00
e0a23bab07
add model_complexity.py
2019-05-22 16:14:04 +01:00
5839a026c3
add person reranking
2019-05-09 22:47:55 +01:00
6342fb3055
check if fpath exists
2019-05-09 22:47:09 +01:00
dee01229da
remove division by 1e+06 when counting model size
2019-04-23 11:34:52 +01:00
aa98ef5a7b
add docstrings to losses, metrics, models, utils, optim
2019-03-22 00:14:41 +00:00
37f92b7cc3
create engine SDK
2019-03-19 17:26:08 +00:00
03d0674dcb
comply woth PEP8
2019-03-15 23:17:38 +00:00
25c64fdd9a
handle UnicodeDecodeError when using python3
2019-03-08 22:27:11 +00:00
4078f18aff
update print msg
2019-02-20 22:28:30 +00:00
5f971cc03a
correct print
2019-02-20 22:22:46 +00:00
c427e88658
rank={:.1%}
2019-02-20 22:20:36 +00:00
d8cfa82594
add print to save_ckpt
2019-02-20 22:17:47 +00:00
19e03d4097
add remove_module_from_keys option to save_ckpt
2019-02-20 22:13:26 +00:00
d56e216068
bug fix
2019-02-20 21:57:09 +00:00
74a2377ab7
add resume_from_ckpt
2019-02-20 21:51:49 +00:00
ae54ca0f6b
mv save_ckpt to torchtools
2019-02-20 21:51:24 +00:00
cf5b9a3ba5
update comments
2019-02-03 14:03:48 +00:00
043c820d01
ignore "module." when saving/loading weights
2019-02-02 14:20:56 +00:00
760d68ebd1
bug fix
2019-02-01 22:43:47 +00:00
2cb1031942
add load_pretrained_weights() to torchtools
2019-02-01 13:10:23 +00:00
800df43fee
check if output is list
2019-02-01 00:50:28 +00:00
564f5a26ab
update print
2019-01-30 22:41:47 +00:00
f045778602
add accuracy
2019-01-27 23:15:38 +00:00
4b6057b923
wrap set_random_seed
2019-01-09 21:30:46 +00:00
074a10e24e
bug fix
2018-12-27 10:25:09 +00:00
KaiyangZhou
Xuanmeng-Zhang
KaiyangZhou
Abner Ayala-Acevedo