301 Commits

Author SHA1 Message Date
kaiyangzhou
f6ba78eceb linting code 2021-08-11 16:59:54 +08:00
kaiyangzhou
b815df6ab0 add osnet_ain x0.75, x0.5 and x0.25 2021-08-11 16:51:32 +08:00
kaiyangzhou
b8b91acbd5 fpath = osp.abspath(osp.expanduser(fpath)) 2021-08-11 16:50:56 +08:00
kaiyangzhou
b882bc8a09 add parse_test_res.py 2021-04-28 16:49:35 +08:00
kaiyangzhou
a379c72adc linting code 2021-04-27 15:47:00 +08:00
kaiyangzhou
c4639e91a0 v1.4.0: added cuhksysu 2021-04-27 15:31:30 +08:00
kaiyangzhou
ea41e69302 RandomIdentitySampler: n_pids should be larger than n_pids_to_sample 2021-04-11 16:04:06 +08:00
Ramakrishnan Sundareswaran
32acbbe4ad
Update mudeep.py to support testing
Current model fails in the case of triplet loss during testing since y,x returns a tuple where engine.py features expects a torch tensor instead to run features.cpu()
2021-04-08 11:31:27 -05:00
kaiyangzhou
c2d9b0e825 update reference 2021-03-26 11:45:06 +08:00
KaiyangZhou
93b8c9f3db v1.3.6: added University-1652 2021-02-14 11:25:24 +08:00
layumi
9b3a836308 add university 2021-02-10 17:59:05 +11:00
Carmel Barak
c03125fb4b Updated version 2021-02-03 15:54:07 +02:00
Carmel Barak
6379253fc8 Merge remote-tracking branch 'origin/master' 2021-02-03 15:45:55 +02:00
Carmel Barak
d78d9a86aa Changed addmm_ to new pytorch API to remove deprecated warnings 2021-02-03 15:45:08 +02:00
Carmel Barak
7cedf08a5b Changed cython types from long to int64_t to support Windows 2021-02-03 15:44:29 +02:00
KaiyangZhou
a105924200 v1.3.4: features.data.cpu() => features.cpu().clone() 2021-01-22 16:21:43 +08:00
KaiyangZhou
454fe67abe avoid huge memory consumption at eval 2021-01-22 16:20:51 +08:00
Xuanmeng-Zhang
9b73eb41d5 Add the GPU-Re-Ranking 2021-01-15 15:29:44 +08:00
KaiyangZhou
626e6c3bb8 minor change in feature_extractor.py 2020-10-19 12:06:30 +01:00
KaiyangZhou
fefa31dc3f 1.3.3: fixed bug in visrank (forgot unpacking dsetid) 2020-08-18 22:17:35 +01:00
KaiyangZhou
208dc3e9b8 add self.feature_dim to osnet 2020-08-18 16:05:58 +01:00
KaiyangZhou
ade7818402 upgrade to v1.3.2 2020-08-12 13:30:01 +01:00
KaiyangZhou
3143997135 add _junk_pids to grid & prid 2020-08-12 13:29:27 +01:00
KaiyangZhou
7c2ee1e6c6 update printing in triplet engine (taking into account weight_t & weight_x) 2020-08-12 11:03:24 +01:00
KaiyangZhou
de42729f41 v1.3.1: record results (rank1 & mAP) of each target dataset 2020-08-10 22:06:15 +01:00
KaiyangZhou
8bbe93ffb5 update version indicator: v1.3.0 2020-08-10 12:39:04 +01:00
KaiyangZhou
4c7c43d34e add RandomDatasetSampler 2020-08-10 12:31:11 +01:00
KaiyangZhou
52b334a63e extend 3-tuple to 4-tuple by adding dsetid 2020-08-10 12:30:54 +01:00
KaiyangZhou
488b9e57db v1.2.7 2020-08-03 16:29:24 +01:00
KaiyangZhou
5656aaba85 bug fix in VideoTripletEngine 2020-08-03 16:17:49 +01:00
KaiyangZhou
069356a6e4 update comment 2020-08-03 16:03:31 +01:00
KaiyangZhou
ad4d8d7c29 v1.2.6: add RandomDomainSampler 2020-08-01 11:52:55 +01:00
KaiyangZhou
b075bd3f4b show lr of last group of params (typically new layers) as default 2020-06-23 18:17:10 +01:00
KaiyangZhou
3478e43148 upgrade to v1.2.5 2020-06-23 15:24:47 +01:00
KaiyangZhou
a5ee40744c add k_tfm 2020-06-23 15:21:11 +01:00
KaiyangZhou
d18b3ad99a v1.2.4 2020-06-23 14:07:54 +01:00
KaiyangZhou
aefe335d68 use dict as return of dataloader 2020-06-23 13:59:57 +01:00
KaiyangZhou
accb3e2146 change scipy.misc.imsave to imageio.imwrite 2020-06-17 22:15:39 +01:00
KaiyangZhou
6081989d7e change fetch_qg() to fetch_test_loaders() 2020-05-05 16:45:22 +01:00
KaiyangZhou
86d97074e5 v1.2.3 (linting code) 2020-05-05 16:06:50 +01:00
KaiyangZhou
5ce43185b1 linting code 2020-05-05 15:58:00 +01:00
KaiyangZhou
d60ca7f224 minor changes 2020-05-05 12:48:58 +01:00
KaiyangZhou
cf091c31f1 v1.2.2 2020-05-05 12:44:08 +01:00
KaiyangZhou
89bf30bd14 update few names of class methods 2020-05-05 12:43:28 +01:00
KaiyangZhou
1708c008ad add argument train_sampler_t (sampler for target train loader) 2020-05-05 11:14:08 +01:00
KaiyangZhou
8e5fc471a6 1.2.1: add simple API for feature extraction 2020-05-04 12:35:45 +01:00
KaiyangZhou
85c446ec97 renaming: loss_dict to loss_summary 2020-04-16 21:57:49 +01:00
KaiyangZhou
b184fba029 save as attribute the vars about steps 2020-04-16 21:41:01 +01:00
KaiyangZhou
b03488b475 upgrade to 1.2.0 2020-04-16 13:15:26 +01:00
KaiyangZhou
36e22e8ce1 make engine more model-agnostic 2020-04-16 12:46:15 +01:00