252 Commits

Author SHA1 Message Date
KaiyangZhou
36e22e8ce1 make engine more model-agnostic 2020-04-16 12:46:15 +01:00
KaiyangZhou
99e719f16e __version__=1.0.9 2019-12-10 10:49:17 +00:00
KaiyangZhou
469a038d01 add scheduler to resume_from_checkpoint 2019-12-10 10:44:45 +00:00
KaiyangZhou
4cc4adb54a add space to make code clearer in transforms.py 2019-12-02 22:27:35 +00:00
KaiyangZhou
188b7f321f fixed missing import in osnet 2019-12-02 22:24:24 +00:00
KaiyangZhou
08bb2cc5eb linting and formatting code 2019-12-02 14:34:59 +00:00
KaiyangZhou
42eaccea71 linting and formatting code 2019-12-02 12:34:17 +00:00
KaiyangZhou
5c88e3664f minor 2019-12-01 03:35:09 +00:00
KaiyangZhou
58e6841a33 update relative import 2019-12-01 03:31:32 +00:00
KaiyangZhou
ac7bd67484 import data, engine, etc. in torchreid/__init__.py 2019-12-01 02:39:29 +00:00
KaiyangZhou
6b8fe56638 linting and formatting code 2019-12-01 02:35:44 +00:00
KaiyangZhou
2095ee05c5 [1.0.8] load_unlabeled_targets -> load_train_targets 2019-11-27 21:20:04 +00:00
KaiyangZhou
7d8431cf49 bug fix for load_unlabeled_targets 2019-11-27 18:19:06 +00:00
KaiyangZhou
41b1e485ec 1.0.7 2019-11-27 17:20:48 +00:00
KaiyangZhou
7e71f72846 give self.writer as argument to engine.train() 2019-11-27 16:52:58 +00:00
KaiyangZhou
86012a4571 add compute_mean_std tool 2019-11-27 16:49:29 +00:00
KaiyangZhou
307e38e519 add use_unlabeled_targets 2019-11-27 16:35:54 +00:00
KaiyangZhou
5c9bac5e75 minor modification 2019-11-27 11:09:13 +00:00
KaiyangZhou
2689fc0c57 rm type-checking for self.model in engine.py 2019-11-24 13:42:46 +00:00
KaiyangZhou
53784903e3 add SequentialSampler 2019-11-24 13:40:54 +00:00
KaiyangZhou
aebb6033e0 add ibn-net 2019-11-10 21:31:28 +00:00
KaiyangZhou
45091f4f85 add __homepage__ 2019-11-04 10:47:49 +00:00
Yao Lu
db00cf5fc3
Fix bugs about dataset of cuhk02.py
Fix bugs about dataset of cuhk02.py.
2019-10-29 10:58:39 +08:00
KaiyangZhou
cab4a342cc v1.0.6 2019-10-23 17:20:34 +01:00
KaiyangZhou
a136384a6a rm unused imports/variables; fixed minor bugs 2019-10-23 17:15:37 +01:00
kaiyangzhou
9397956920 update version number to 1.0.5 2019-10-22 23:06:25 +01:00
kaiyangzhou
edf0b73e21 add OSNet-AIN code 2019-10-22 22:48:53 +01:00
kaiyangzhou
fad38231c1 add radam optimizer 2019-10-22 22:47:08 +01:00
kaiyangzhou
894de59bff add new ref to osnet.py 2019-10-22 22:46:30 +01:00
kaiyangzhou
1a5db72550 update annotation in RandomPatch 2019-10-22 22:45:48 +01:00
KaiyangZhou
17887c05fd version=1.0.3 2019-10-04 12:15:32 +01:00
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