mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
482 Commits
1 Branch
4 Tags
15 MiB
Commit Graph

482 Commits (bb6ddbf8b15520bae9090827f0ab3dc8cf3b1337)
 

Author SHA1 Message Date
Xingyu Liao bb6ddbf8b1
bugfix for visualize and demo (#468) 
Summary minor bugs for visualize and demo caused by revised DataLoader
 2021-04-21 16:24:34 +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
Darren e0ad8c70bc
update dockerfile in fastrt (#437) 
Reviewed by: @L1aoXingyu
 2021-04-12 15:06:13 +08:00
慕湮 fc67350e99
Add python interface by pybind11 and Int8 mode 
Reviewed by: @TCHeish
 2021-04-12 15:05:21 +08:00
Xingyu Liao 1dce15efad
faster dataloader with pre-fetch and cuda stream (#456) 
Summary: add a background thread to create a generator with pre-fetch, and create a new cuda stream to copy tensor from cpu to gpu in parallel.

Reviewed by: l1aoxingyu
 2021-04-12 15:03:35 +08:00
liaoxingyu 0da5917064 bugfix for ClasDataset 2021-04-08 11:01:15 +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
liaoxingyu 9288db6303 add filp in fasttune config 2021-03-31 17:09:34 +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
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 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 664ba4ae11
save idx2class dict when training classification (#444) 2021-03-26 20:51: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 9d83550b67 bugfix 
Summary: make `deploy` adaptation for other models except repvgg
 2021-03-26 10:10:45 +08:00
Xingyu Liao 15c556c43a
remove apex dependency (#442) 
Summary: Use Pytorch1.6(or above) built-in amp training
 2021-03-23 12:12:35 +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
Xingyu Liao 41c3d6ff4d
add Dockerfile (#440) 2021-03-23 11:45:03 +08:00
Xingyu Liao 9b5af4166e
support repvgg (#429) 
Summary:
* support repvgg backbone, and verify the consistency of train mode and eval mode
* onnx export logger style modification
 2021-03-11 14:23:24 +08:00
Xingyu Liao cb7a1cb3e1
update deployment toolchain (#428) 
Summary:
Remove tiny-tensorrt dependency and rewrite a new tensorrt inference api.
In the new version of trt infer, it can pad the input to fixed batch automatically, so you don't need to worry about dynamic batch size.
 2021-03-10 16:48:59 +08:00
Darren d7c1294d9e
fastrt patch update 
Summary:
move div255 to gpu
add read/write numpy ndarray, which will make the comparison between torch and trt results more easily.

Reviewed By: l1aoxingyu
 2021-03-10 10:53:18 +08:00
Xingyu Liao 0cc9fb95a6
Merge pull request #425 from JDAI-CV/multi-node 
Summary: Add multiple machine training getting started docs.
Change multiple dataset evaluation logging mode, which will show the testing result of each dataset immediately.

Reviewed by: l1aoxingyu
 2021-03-09 20:13:29 +08:00
liaoxingyu f57c5764e3 support multi-node training 2021-03-09 20:07:28 +08:00
liaoxingyu 68c190b53c replace list in evaluator process with dict 2021-03-09 20:07:13 +08:00
Xingyu Liao 44ad4b83b1
update veriwild training config 
Summary: fix veriwild training config for reproducing.
Reviewed by: l1aoxingyu
 2021-03-04 15:20:39 +08:00
Xingyu Liao fcfa6800bb
Merge pull request #416 from TCHeish 
Reviewed by: l1aoxingyu
 2021-03-04 15:18:47 +08:00
Xinchen Liu 575aeaec3f
Update bagtricks_R50-ibn.yml 2021-03-03 12:07:39 +08:00
darrenhsieh b9bda486f0 add trt api div255 2021-02-27 16:45:29 +08:00
darrenhsieh 69eb044b81 [v004] de-couple fastreid config from Model base-class 
fix baseline class ctr bug
 2021-02-27 16:40:04 +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 e2a1e14bc3 remove extra info 2021-02-24 14:40:45 +08:00
Xingyu Liao 2f1836825c
update requirements.txt location 2021-02-23 19:05:24 +08:00
Namgyu Ho 527b09c696
Update requirement.txt location in INSTALL.md 
Previously, it was hard to find the requirements.txt file
 2021-02-23 14:50:16 +09:00
liaoxingyu c4412d367f update GETTING_STARTED.md 2021-02-18 11:34:45 +08:00
liaoxingyu 819c5a8ab4 update vehicle reid config 2021-02-18 11:33:05 +08:00
Xingyu Liao cf46e5f071
Attribute projects update 
Summary: 1.fix zero divided in attribute metric computation;2.update market/duke attribute dataset loading.
Reviewed by: l1aoxingyu
 2021-02-18 10:32:48 +08:00
Xingyu Liao 3854069f4e
update datset description 
Reviewed by: l1aoxingyu
 2021-02-18 10:27:50 +08:00
Xingyu Liao 2f95da9a59
patch update for fasttrv003 2021-02-18 10:26:55 +08:00
darrenhsieh 0617e3eeb7 refactor: embedding_head 
* add abstract poolinglayer

* add poolinglayer factory
 2021-02-13 21:28:29 +08:00
darrenhsieh 159494e4a4 refactor: model class 
* add Module as base class

* add head type

* replace template func args with Module ptr

* add Module factory
 2021-02-13 21:14:14 +08:00
Zhedong Zheng 96b7d3133a
Update README.md 2021-02-10 15:29:09 +08:00
Jinkai Zheng 39e25e0457 add Market1501Attr and DukeMTMCAttr dataloader 2021-02-05 22:36:54 +08:00
Jinkai Zheng c9537c97d1 add eps in attr_evaluation.py 2021-02-05 21:52:39 +08:00
liaoxingyu 254a489eb1 Merge branch 'master' of github.com:L1aoXingyu/fast-reid 2021-02-01 10:56:24 +08:00
liaoxingyu 50ceeb0832 add gitter url 2021-02-01 10:55:24 +08:00
Xingyu Liao a8ae0fd1e9
fastRT refactor 
fix cmake bug and model.h refacotr
 2021-02-01 10:12:22 +08:00
darrenhsieh 5f7d3d586e refactor: model.h 2021-01-31 18:12:35 +08:00