haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix a bug with maskrcnn (#758)

Browse Source
This commit is contained in:
JumpPandaer 2021-10-13 19:04:39 +08:00 committed by GitHub
parent 7c1a145c34
commit c70a338827
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rcnn/MaskRcnnInferencePlugin.h
View File

@ -25,7 +25,7 @@ int maskRcnnInference(int batchSize,
class MaskRcnnInferencePlugin : public IPluginV2Ext {
int _detections_per_im;
int _output_size;
int _num_classes;
int _num_classes = 1;
protected:
void deserialize(void const* data, size_t length) {

2
rcnn/README.md
View File

@ -124,7 +124,7 @@ 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.
- do initializing for _pre_nms_topk in RpnNmsPlugin, _count in BatchedNmsPlugin and _num_classes in MaskRcnnInferencePlugin 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