Commit Graph

33 Commits (master)

Author SHA1 Message Date
liaoxingyu 764fa67fe9 Align bagtricks config with v1.2 2021-06-01 11:51:22 +08:00
liaoxingyu 07b8251ccb Support gradient clip
Follow detectron2's instruction and add gradient clip in step function of optimizer
2021-05-31 17:11:37 +08:00
liaoxingyu 2cabc3428a Support vision transformer backbone 2021-05-31 17:08:57 +08:00
liaoxingyu 55300730e1 update fastreid v1.2 readme and changelog 2021-04-06 20:09:13 +08:00
liaoxingyu 44cee30dfc update fastreid v1.2
Summary:
1. refactor dataloader and heads
2. bugfix in fastattr, fastclas, fastface and partialreid
3. partial-fc supported in fastface
2021-04-02 21:33:13 +08:00
Xingyu Liao fb36b23678
bugfix for attribute project (#450)
Summary: refactor sample weight in attribute recognition;
change all options to False in defaults.py and modify yaml files
2021-03-31 17:07:19 +08:00
Xingyu Liao be0a089e1f
bugfix & merge classification transforms (#448)
Summary: change heads definition in project and config file, merge classification transforms into default transforms
2021-03-30 15:47:14 +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
Xingyu Liao 883fd4aede
add configurable decorator & linear loss decouple (#441)
Summary: Add configurable decorator which can call `Baseline` with `Baseline(cfg)` or `Baseline(cfg, heads=heads, ...)`
Decouple linear and loss computation for partial-fc support.

Reviewed By: l1aoxingyu
2021-03-23 12:10:06 +08:00
Xinchen Liu 575aeaec3f
Update bagtricks_R50-ibn.yml 2021-03-03 12:07:39 +08:00
liaoxingyu 819c5a8ab4 update vehicle reid config 2021-02-18 11:33:05 +08:00
Xinchen Liu 885dc96608
Update bagtricks_R50-ibn.yml
The config that obtains the best performance on VeRi-Wild. The model is trained on 4 P40 GPUs
2021-01-28 11:15:57 +08:00
liaoxingyu e26182e6ec make lr warmup by iter
Summary: change warmup way by iter not by epoch, which will make it more flexible when training small epochs
2021-01-22 11:17:21 +08:00
liaoxingyu 15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu 130199fdce remove invalid config 2020-12-07 14:23:43 +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 3c48bf78c1 change config file 2020-09-10 10:58:56 +08:00
liaoxingyu 78f2166596 refactor mgn config 2020-09-02 18:01:07 +08:00
liaoxingyu 0d2c477060 update mgn config 2020-09-01 16:16:40 +08:00
liaoxingyu 64861ea6c8 update config file 2020-07-29 19:48:49 +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 ea8a3cc534 fix typro 2020-07-10 16:26:35 +08:00
liaoxingyu 59a74de240 update vehicle reid config 2020-07-09 12:10:38 +08:00
liaoxingyu 825da4dca0 update all config files 2020-07-06 16:48:14 +08:00
liaoxingyu eebc71b7c8 update config
Summary: remove use_cosine_dist in config which is equivalent to euclidean distance with norm feature
2020-06-16 11:14:58 +08:00
Jinkai Zheng 5fb5e771e9 Add vehicle configs 2020-05-29 16:06:04 +08:00
liaoxingyu e20eb0394b fix: modify open_layers in BoT config file
modify open_layers to list type to be compatible with default config
2020-05-19 14:46:35 +08:00
liaoxingyu 9e40194669 fix: change config of sbs
change weight decay and bias_lr in config of SBS, which can boost performance slightly.
2020-05-18 20:16:38 +08:00
liaoxingyu 18a33f7962 feat: add MGN model
support MGN architecture and training config
2020-05-15 11:39:54 +08:00
liaoxingyu 3041a52aef fix: revise config file in stronger baseline 2020-05-14 14:55:30 +08:00
liaoxingyu 46228ce946 chore(configs): update all training config 2020-05-01 09:04:51 +08:00
liaoxingyu 2327a5565f chore(configs): update Market1501 training config 2020-04-29 21:06:57 +08:00
liaoxingyu ec19bcc1d3 style(configs): put all config files together
put all config files into one place for easily control,
and add tools for put train_net.py which almost the same in
different projects
2020-04-29 16:18:54 +08:00