parent
101e60348e
commit
7b85cfd392
@ -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) {
|
||||
|
||||
@ -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.
|
||||
|
||||
@ -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:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user