see note in README
This commit is contained in:
JumpPandaer 2021-06-28 10:31:59 +08:00 committed by GitHub
parent 101e60348e
commit 7b85cfd392
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 2 deletions

View File

@ -30,7 +30,7 @@ class BatchedNmsPlugin : public IPluginV2Ext {
float _nms_thresh;
int _detections_per_im;
size_t _count;
size_t _count = 1;
protected:
void deserialize(void const* data, size_t length) {

View File

@ -124,6 +124,8 @@ sudo ./rcnn -d faster.engine ../samples
- you can build fasterRcnn with maskRcnn weights file.
- do initializing for _pre_nms_topk in RpnNmsPlugin and _count in BatchedNmsPlugin inside class to prevent error assert, because the configurePlugin function is implemented after clone() and before serialize(). one can also set it through constructor.
## Quantization
1. quantizationType:fp32,fp16,int8. see BuildRcnnModel(rcnn.cpp line 276) for detail.

View File

@ -28,7 +28,7 @@ class RpnNmsPlugin : public IPluginV2Ext {
float _nms_thresh;
int _post_nms_topk;
size_t _pre_nms_topk;
size_t _pre_nms_topk = 1;
mutable int size = -1;
protected: