Commit Graph

271 Commits (069356a6e4915fa01b7abb51a15b5dfd816db029)

Author SHA1 Message Date
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