Skip to content

Commit 342dffe

Browse files
authored
Merge pull request #18 from andife/patch-2
Update webcam_inference.py
2 parents dda4c85 + 4ba4a7a commit 342dffe

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

webcam_inference.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,12 @@
22
import numpy as np
33
import cv2
44

5-
65
from detr_tf.training_config import TrainingConfig, training_config_parser
76
from detr_tf.networks.detr import get_detr_model
87
from detr_tf.data import processing
98
from detr_tf.data.coco import COCO_CLASS_NAME
109
from detr_tf.inference import get_model_inference, numpy_bbox_to_image
1110

12-
1311
@tf.function
1412
def run_inference(model, images, config):
1513
m_outputs = model(images, training=False)
@@ -57,5 +55,5 @@ def run_webcam_inference(detr):
5755
detr = get_detr_model(config, include_top=True, weights="detr")
5856
config.background_class = 91
5957

60-
# RUn webcam inference
61-
run_webcam_inference(detr)
58+
# Run webcam inference
59+
run_webcam_inference(detr)

0 commit comments

Comments
 (0)