mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity
785 Commits 3 Branches 20 Tags
Commit Graph

56 Commits

Author SHA1 Message Date
KaiyangZhou
36e22e8ce1 make engine more model-agnostic 2020-04-16 12:46:15 +01:00
KaiyangZhou
469a038d01 add scheduler to resume_from_checkpoint 2019-12-10 10:44:45 +00:00
KaiyangZhou
08bb2cc5eb linting and formatting code 2019-12-02 14:34:59 +00:00
KaiyangZhou
6b8fe56638 linting and formatting code 2019-12-01 02:35:44 +00:00
KaiyangZhou
a136384a6a rm unused imports/variables; fixed minor bugs 2019-10-23 17:15:37 +01:00
kaiyangzhou
7a73dca4ac bug fix: img_path->path 2019-09-13 23:23:13 +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
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
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
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
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
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
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
dee01229da remove division by 1e+06 when counting model size 2019-04-23 11:34:52 +01: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
25c64fdd9a handle UnicodeDecodeError when using python3 2019-03-08 22:27:11 +00:00
KaiyangZhou
4078f18aff update print msg 2019-02-20 22:28:30 +00:00
KaiyangZhou
5f971cc03a correct print 2019-02-20 22:22:46 +00:00
KaiyangZhou
c427e88658 rank={:.1%} 2019-02-20 22:20:36 +00:00
KaiyangZhou
d8cfa82594 add print to save_ckpt 2019-02-20 22:17:47 +00:00
KaiyangZhou
19e03d4097 add remove_module_from_keys option to save_ckpt 2019-02-20 22:13:26 +00:00
KaiyangZhou
d56e216068 bug fix 2019-02-20 21:57:09 +00:00
KaiyangZhou
74a2377ab7 add resume_from_ckpt 2019-02-20 21:51:49 +00:00
KaiyangZhou
ae54ca0f6b mv save_ckpt to torchtools 2019-02-20 21:51:24 +00:00
KaiyangZhou
cf5b9a3ba5 update comments 2019-02-03 14:03:48 +00:00
KaiyangZhou
043c820d01 ignore "module." when saving/loading weights 2019-02-02 14:20:56 +00:00
KaiyangZhou
760d68ebd1 bug fix 2019-02-01 22:43:47 +00:00
KaiyangZhou
2cb1031942 add load_pretrained_weights() to torchtools 2019-02-01 13:10:23 +00:00
KaiyangZhou
800df43fee check if output is list 2019-02-01 00:50:28 +00:00
KaiyangZhou
564f5a26ab update print 2019-01-30 22:41:47 +00:00
KaiyangZhou
f045778602 add accuracy 2019-01-27 23:15:38 +00:00
KaiyangZhou
4b6057b923 wrap set_random_seed 2019-01-09 21:30:46 +00:00
KaiyangZhou
074a10e24e bug fix 2018-12-27 10:25:09 +00:00
KaiyangZhou
54b92ee617 check if isinstance(model, nn.DataParallel) 2018-11-08 23:54:26 +00:00
KaiyangZhou
205df55987 add fixbase functions (debug mode) 2018-11-08 23:04:50 +00:00
KaiyangZhou
c32cf80a9f add RankLogger 2018-11-08 21:38:22 +00:00
KaiyangZhou
fdc451c0e5 add linear decay 2018-11-07 22:02:23 +00:00
KaiyangZhou
1f1f01669b change dataset to (query, gallery) 2018-11-07 21:45:48 +00:00
KaiyangZhou
3bd674f198 use imgname as dirname for each query 2018-10-06 22:21:11 +01:00