We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents dda4c85 + 4ba4a7a commit 342dffeCopy full SHA for 342dffe
webcam_inference.py
@@ -2,14 +2,12 @@
2
import numpy as np
3
import cv2
4
5
-
6
from detr_tf.training_config import TrainingConfig, training_config_parser
7
from detr_tf.networks.detr import get_detr_model
8
from detr_tf.data import processing
9
from detr_tf.data.coco import COCO_CLASS_NAME
10
from detr_tf.inference import get_model_inference, numpy_bbox_to_image
11
12
13
@tf.function
14
def run_inference(model, images, config):
15
m_outputs = model(images, training=False)
@@ -57,5 +55,5 @@ def run_webcam_inference(detr):
57
55
detr = get_detr_model(config, include_top=True, weights="detr")
58
56
config.background_class = 91
59
60
- # RUn webcam inference
61
- run_webcam_inference(detr)
+ # Run webcam inference
+ run_webcam_inference(detr)
0 commit comments