small typo in readme
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ inputs = processor(images=image, return_tensors="pt")
|
|
37 |
outputs = model(**inputs)
|
38 |
|
39 |
# convert outputs (bounding boxes and class logits) to COCO API
|
40 |
-
# let's only keep detections with score > 0.
|
41 |
target_sizes = torch.tensor([image.size[::-1]])
|
42 |
results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.8)[0]
|
43 |
|
|
|
37 |
outputs = model(**inputs)
|
38 |
|
39 |
# convert outputs (bounding boxes and class logits) to COCO API
|
40 |
+
# let's only keep detections with score > 0.8
|
41 |
target_sizes = torch.tensor([image.size[::-1]])
|
42 |
results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.8)[0]
|
43 |
|