18 Commits

Author SHA1 Message Date
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
dbf1604231 fix model deploy problems 2021-05-25 15:55:43 +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
liaoxingyu
9d83550b67 bugfix
Summary: make `deploy` adaptation for other models except repvgg
2021-03-26 10:10:45 +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
liaoxingyu
e2a1e14bc3 remove extra info 2021-02-24 14:40:45 +08:00
liaoxingyu
7e83d3175f update README.md
Summary: add information about fastreid V1.0
2021-01-18 11:44:55 +08:00
liaoxingyu
15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu
07872d4cdb fix caffe export bug
Summary: put tensor to gpu device
2020-08-20 16:28:52 +08:00
liaoxingyu
64861ea6c8 update config file 2020-07-29 19:48:49 +08:00
liaoxingyu
a8c7e07df6 update deploy docs 2020-07-29 17:47:35 +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
fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu
27c48c8f02 update pytorchtocaffe tool
Summary: update pytorch to caffe toolchain to support model layers and bash scripts
2020-06-09 14:45:27 +08:00
Christian Clauss
d7cf32a652
Fix a Python syntax error discovered by #77 2020-06-03 14:32:46 +02:00
liaoxingyu
69e4ba901e update acknowledgement in deploy 2020-06-02 18:12:57 +08:00
liaoxingyu
2f6d999469 add pytorch to caffe converting
Summary: update deployment README and support pytorch to caffe converting for basemodel
2020-06-02 18:11:02 +08:00