liaoxingyu
10b04b75ff
Support lazy loading in dataset
...
Add an option for lazy loading, which will parse the data when using it. Specifically, when calling `dataset.query`, it will start to parse the query data, and `train` and `gallery` will not be parsed in advanced. This is useful when you want to run on test dataset but train dataset is large.
See examples in market1501.py
2021-06-17 16:59:02 +08:00
liaoxingyu
d792a69f3f
bugfix for veriwild and vehicleid
...
Change `camids` from string to int when evaluation, otherwise, it will raise AttributeError
#497
2021-06-17 16:52:47 +08:00
Jinkai Zheng
f2286e7f55
Updated the way of reading VERI-Wild's TXT files ( #507 )
...
Reviewed by: @L1aoXingyu
2021-06-11 13:46:46 +08:00
Xie Jingyi
25cfa88fd9
Fix bug: COMBINEALL should add extra prefix to avoid pid conflictions ( #447 )
2021-03-30 15:49:52 +08:00
Xingyu Liao
890224f25c
support classification in fastreid ( #443 )
...
Summary: support classification and refactor build_dataloader which can support explicit parameters passing
2021-03-26 20:17:39 +08:00
liaoxingyu
96fd58c48f
update vehicleid and veriwild datasets
2021-02-26 19:26:48 +08:00
liaoxingyu
52b75b7974
update prid and grid datasets
2021-02-26 19:26:38 +08:00
liaoxingyu
a53fd17874
update docs
2021-01-23 15:25:58 +08:00
liaoxingyu
b5c3c0a24d
update docs
2021-01-22 21:11:19 +08:00
Xingyu Liao
0428a270e3
update veri dataset
...
Summary: update veri dataset reference information
Reviewed By: l1aoxingyu
2020-12-22 18:01:37 +08:00
liaoxingyu
04fe9fb2d8
add saivt dataset
2020-12-22 15:47:08 +08:00
liaoxingyu
e8739a667e
update AirportALERT dataset
...
Summary: add more IDs from the single camera
2020-12-22 15:36:50 +08:00
Xinchen Liu
218a007b6a
Update veri.py
2020-12-21 16:19:06 +08:00
liaoxingyu
a327a70f0d
v0.3 update
...
Summary:
1. change DPP training in apex way;
2. make warmup scheduler by iter and lr scheduler by epoch;
3. replace random erasing with torchvision implementation;
4. naming modification in config file
2020-12-07 14:19:20 +08:00
liaoxingyu
10cbaab155
support finetuning from trained models
...
Summary: add a flag for supporting finetuning model from the trained weights, and it's very useful when performing across domain reid
2020-09-28 17:10:10 +08:00
liaoxingyu
3024cea3a3
add more datasets
...
Summary: add wildtracker.py datasets and cuhk_sysu.py datasets.
2020-09-28 17:05:40 +08:00
liaoxingyu
154a06b875
refactor code
2020-09-23 19:31:46 +08:00
liaoxingyu
919a515eb7
fix combine all training
2020-09-09 15:28:21 +08:00
liaoxingyu
866a196d19
add more datasets support
2020-09-01 16:13:12 +08:00
liaoxingyu
9c667d1a0f
add pretrain_path support in backbone
2020-08-14 14:00:26 +08:00
liaoxingyu
d1c20cbe50
fix pre-train model bugs
...
fix bugs locks when downloading pre-train model
2020-08-04 15:56:36 +08:00
liaoxingyu
977aa4a357
update model url
2020-07-31 14:59:37 +08:00
liaoxingyu
65169b40bd
support ddp testing
2020-07-30 20:15:28 +08:00
liaoxingyu
16655448c2
onnx/trt support
...
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu
c851c7edc8
fix dataset prefix #76
2020-07-06 16:54:54 +08:00
liaoxingyu
8879db3fba
update training instruction
...
Summary: update dataset configuration and training instruction
2020-06-16 19:43:36 +08:00
liaoxingyu
fd90555e19
feat: add multi-dataset joint training
...
new feature that can support joint training, and find some bugs in funtion combine_all of datasets/bases.py that assume person id in dataset has been relabeld from 0 to num_class.
Another bug appears in msmt17 which trainset and testset person id both begin from 0, and we should change testset id from num_class of trainset.
2020-05-18 20:06:04 +08:00
Jinkai Zheng
640c9bfc97
Add small, medium and large vehicle test datasets
2020-05-11 01:34:19 -05:00
Jinkai Zheng
e059b751a6
Add modified vehicle datasets with small, medium and large test datastes
2020-05-11 01:30:56 -05:00
zjk15068083791
2f3f6e3267
Add files via upload
2020-04-27 16:27:44 +08:00
liaoxingyu
bb50b6c5a7
docs($projects): update agw readme
2020-04-21 11:35:54 +08:00
liaoxingyu
9684500a57
chagne arch
...
1. change dataset show to trainset show and testset show seperately
2. add cls layer to easily plug in circle loss and arcface
2020-04-19 12:54:01 +08:00
liaoxingyu
be9faa5605
update focal loss
...
update dataset info display
update seperate lr
update adaptive label smooth regularization
2020-04-17 13:46:10 +08:00
liaoxingyu
6a8961ce48
1. upload circle loss and arcface
...
2. finish freeze training
3. update augmix data augmentation
2020-04-05 23:54:26 +08:00
L1aoXingyu
8a9c0ccfad
Finish first version for fastreid
2020-02-10 22:13:04 +08:00
L1aoXingyu
db6ed12b14
Update sampler code
2020-02-10 07:38:56 +08:00
liaoxingyu
71950d2c09
1. Fix evaluation code
...
2. Finish multi-dataset evaluation
3. Decouple image preprocess and output postprocess with model forward for DataParallel training
4. Finish build backbone registry
5. Fix dataset sampler
2020-01-21 20:24:26 +08:00
liaoxingyu
b761b656f3
Finish basic training loop and evaluation results
2020-01-20 21:33:37 +08:00