ZXYFrank
94d6b02e40
Update datamanager.py
...
add **kwargs to all occurrences of `init_image_dataset ` in `datamanager.py`
2022-02-23 21:23:47 +08:00
ZXYFrank
a6b2f77e8a
Add **kwargs to ImageDataManager
...
Add **kwargs optional arguments to ImageDataManager
I defined a custom ImageDataset, which takes some additional arguments for init.
According to [use custom dataset](https://kaiyangzhou.github.io/deep-person-reid/user_guide.html#use-your-own-dataset ), the officially recommended way to use the dataset is to use the ImageDataManager
I tried to pass **kwargs(a dict) into the torchreid.data.ImageDataManager() constructor. Unfortunately, due to the **positional(fixed)** arguments of the function `init_image_dataset` (in datamanager.py), the constructor of my own custom dataset cannot receive any additional arguaments.
I hope it will fix the problem!
2022-02-23 20:37:42 +08:00
ZXYFrank
07709e7ab0
Add optional arguments to ImageDataManager
...
Add **kwargs optional arguments to ImageDataManager, which will ensure flexibility of the custom and registered dataset
2022-02-23 20:36:33 +08:00
KaiyangZhou
4c7c43d34e
add RandomDatasetSampler
2020-08-10 12:31:11 +01:00
KaiyangZhou
ad4d8d7c29
v1.2.6: add RandomDomainSampler
2020-08-01 11:52:55 +01:00
KaiyangZhou
a5ee40744c
add k_tfm
2020-06-23 15:21:11 +01:00
KaiyangZhou
6081989d7e
change fetch_qg() to fetch_test_loaders()
2020-05-05 16:45:22 +01:00
KaiyangZhou
5ce43185b1
linting code
2020-05-05 15:58:00 +01:00
KaiyangZhou
1708c008ad
add argument train_sampler_t (sampler for target train loader)
2020-05-05 11:14:08 +01: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
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
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
5365619606
add config
2019-08-26 10:34:31 +01:00
KaiyangZhou
27d4f2f093
add visactmap code; upgrade visrank to plot single figure
2019-08-03 17:35:06 +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
709018d948
add argument 'transforms' to datamanager
2019-07-03 12:05:55 +01:00
KaiyangZhou
e5b8c464ca
update docstrings
2019-03-24 23:33:54 +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