New features

- Added support for INT8 calibration
- Added support for non square models
- Updated mAP comparison between models
This commit is contained in:
Marcos Luciano
2021-06-18 00:30:10 -03:00
parent 312e9a448d
commit cbd9675dc2
74 changed files with 3287 additions and 700 deletions

View File

@@ -4,10 +4,11 @@ net-scale-factor=0.0039215697906911373
model-color-format=0
custom-network-config=pgie/yolo.cfg
model-file=yolo.weights
model-engine-file=model_b1_gpu0_fp16.engine
model-engine-file=model_b1_gpu0_fp32.engine
#int8-calib-file=calib.table
labelfile-path=labels.txt
batch-size=1
network-mode=2
network-mode=0
num-detected-classes=2
interval=0
gie-unique-id=1