Add variable batch for detector update

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-19 09:32:27 +02:00
parent ad21cc0b79
commit f2991bec5c
3 changed files with 10 additions and 7 deletions
@@ -111,7 +111,7 @@ int main(int argc, char *argv[]) {
            break;
    
        //inference
-        detNN->update(batch_dnn_input);
+        detNN->update(batch_dnn_input, n_batch);
        detNN->draw(batch_frame);

        if(show){