Commit Graph

170 Commits (bc63002d05b2ebdedc3397f0c6a409a2a562729d)

Author SHA1 Message Date
zuchen.wang bc63002d05 update online-pcb.yaml 2021-11-12 19:39:40 +08:00
zuchen.wang 781b989407 过滤0张或者1张图像 2021-11-12 15:25:43 +08:00
zuchen.wang 93edc0cf53 change evaluator 2021-11-10 17:01:27 +08:00
zuchen.wang f8ae876c8c fix: 删除dataset相关的日志中令人迷惑的csv format 2021-11-10 17:00:50 +08:00
zuchen.wang 1d8451a23a fix contrastive loss 2021-11-04 16:24:18 +08:00
zuchen.wang 3d0ac7dc6b rm useless config yaml 2021-11-04 10:53:53 +08:00
Sherlock 1342f5d599 add interpreter project 2021-11-04 10:41:36 +08:00
zuchen.wang 01538bec64 fix logger and pretty dataset info 2021-11-04 10:11:16 +08:00
zuchen.wang 21c14f1494 重构数据集 2021-11-03 17:25:58 +08:00
zuchen.wang 65cfc515d9 实现和线上模型一模一样的PCBNet 2021-11-02 17:56:05 +08:00
zuchen.wang 5a075c1fe8 Merge branch 'develop' into pcb_online 2021-10-29 14:29:29 +08:00
zuchen.wang d03eae9daa Implement online pcb config 2021-10-29 14:13:45 +08:00
zuchen.wang c90792042a make common base config: input, dataloader, solver, test period and batch 2021-10-29 11:45:24 +08:00
zuchen.wang eabe896ee8 train pcb, 未达到期望 2021-10-28 22:39:54 +08:00
zuchen.wang ad31a2cd16 modify config 2021-10-27 16:15:58 +08:00
zuchen.wang 3880b23ab4 refactor evaluator 2021-10-27 16:03:43 +08:00
zuchen.wang d4e2ac32d8 add pcb 2021-10-27 15:09:57 +08:00
zuchen.wang 31d7a51b62 fix bug in online_dataset.py 2021-10-22 13:31:12 +08:00
zuchen.wang ae842114c7 update dataset config in trainer.py 2021-10-21 10:50:38 +08:00
zuchen.wang 0dd493b7b4 make pair dataset negative sample from full set 2021-10-20 17:04:27 +08:00
zuchen.wang 6ab4ac9784 modify logger 2021-10-18 17:38:08 +08:00
zuchen.wang e4c46cbfed add r101-se-ibn.yaml 2021-10-18 17:37:41 +08:00
zuchen.wang 3788c5484b add excel data set 2021-10-18 15:37:25 +08:00
zuchen.wang 3e753612a9 change pair evaluator 2021-10-18 15:36:54 +08:00
zuchen.wang 7190348568 fix pair_dataset random seed in val and test mode
add ce loss
2021-10-16 21:42:32 +08:00
zuchen.wang 1feda07ce6 evaluation success 2021-10-16 21:24:16 +08:00
zuchen.wang 786c442391 add bughook in train_net.py 2021-10-16 20:56:01 +08:00
zuchen.wang d87f4a8419 add evaluator in fastshoe trainner 2021-10-16 00:03:00 +08:00
zuchen.wang d162c5fc4e add PairHead 2021-10-15 21:26:16 +08:00
zuchen.wang 8c09ec0d3d update forward 2021-10-13 20:05:08 +08:00
zuchen.wang d2ee44d503 change cls_target to clas_target 2021-10-13 20:05:08 +08:00
zuchen.wang 1bd41a79fc add cls target in pair dataset 2021-10-13 20:05:07 +08:00
zuchen.wang 8b309b0f4e add contrastive loss 2021-10-13 20:05:07 +08:00
zuchen.wang 7201a82840 add pair collator_fn 2021-10-13 20:05:07 +08:00
zuchen.wang bced0d04ff override some method in dataset 2021-10-13 20:05:07 +08:00
zuchen.wang e9cd5311f5 add some paths 2021-10-13 20:05:07 +08:00
zuchen.wang 46c1ce83a7 add fastshoe 2021-10-13 20:05:07 +08:00
zuchen.wang 045b1a88bc fix import error in fastclas
add bughook.py
2021-10-13 18:07:44 +08:00
l1aoxingyu f4551a128b feat(projects): Add DG-ReID project 2021-09-08 23:10:47 +08:00
xiaomingzhid ee3818e706
Add Semi-Supervised Domain Generalizable Person Re-Identification 2021-09-08 21:11:32 +08:00
Sherlock ced654431b fix bug in attribute evaluation
# 535
2021-08-12 14:17:44 +08:00
liaoxingyu 7e652fea2a feat: Add contiguous parameters support
Support contiguous parameters to train faster. It can split parameters into different contiguous groups by freeze_layer, lr and weight decay.
2021-07-05 11:10:37 +08:00
liaoxingyu 7c1269d6c3 minor fix for configs
#519
2021-06-24 14:26:51 +08:00
liaoxingyu 2fee8327fa minor fix for configs
#519
2021-06-23 17:49:35 +08:00
liaoxingyu c3ac4f504c Support amp and resume training in fastface
AMP in partial-fc needs to be done only on backbone; In order to impl `resume training`, need to save & load different part of classifier weight in each GPU.
2021-05-31 17:30:43 +08:00
liaoxingyu 91ff631184 Minor changes
Some minor changes, such as class name changing, remove extra blank line, etc.
2021-05-31 17:27:14 +08:00
liaoxingyu 8ab3554958 Support self-distill with EMA updated model 2021-05-31 17:17:24 +08:00
Darren 37ccd3683d
unify gen_wts.py and inference.cpp dummy test value. (#457)
Reviewed by: @L1aoXingyu
2021-04-19 10:23:15 +08:00
Darren 8276ccf4fd
[v005] set INT8 calibrate set via cmake (#459)
Reviewed by: @L1aoXingyu
2021-04-19 10:22:48 +08:00
liaoxingyu e124a9afd3 fix lint_python 2021-04-12 15:08:39 +08:00