Commit Graph

305 Commits (master)

Author SHA1 Message Date
HanCheng be55021c57
Update tools.py
If you don't use the condition expression speed may be 0 initially

deep-person-reid-master\torchreid\utils\tools.py", line 86, in _reporthook
    speed = int(progress_size / (1024*duration))
ZeroDivisionError: float division by zero
2022-08-15 16:43:02 +08:00
Kaiyang e34e3ae85f
Merge pull request #493 from Bloodrammer/master
Fix memory leak in Engine
2022-02-24 00:13:56 +08:00
Bloodrammer 0e82980e8a Fix memory leak in Engine 2022-02-22 17:25:54 +03:00
Helmut Grohne 42185c4beb FeatureExtractor: avoid loading weights twice
When passing a model_path to the FeatureExtractor, weights are loaded twice.
One the pretrained once and then the given ones. Avoid the unnecessary load.
2022-01-19 09:25:20 +01:00
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