Bounding box detection
PyTorch
huu-ontocord commited on
Commit
a62447c
1 Parent(s): 2b4def6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,5 +26,5 @@ box_segmentation_model= GeneralizedRCNN.from_pretrained("unc-nlp/frcnn-vg-finetu
26
 
27
  img_url = 'image.png'
28
  raw_image = Image.open(img_url).convert('RGB')
29
- frcnn_output = decode_image(np.asarray(raw_image), frcnn, image_preprocessor, max_detections=max_detections)
30
  ```
 
26
 
27
  img_url = 'image.png'
28
  raw_image = Image.open(img_url).convert('RGB')
29
+ frcnn_output = decode_image(np.asarray(raw_image), box_segmentation_model, image_preprocessor, max_detections=max_detections)
30
  ```