Update README.md

README.md

@@ -2,20 +2,18 @@
 license: apache-2.0
 ---
 ```
-from frcnn.frcnn_ids import *
-from frcnn.visualizing_image import SingleImageViz
-from frcnn.processing_image import *
-from frcnn.modeling_frcnn import *
-from frcnn.utils import *
-  
 
+from frcnn.visualizing_image import SingleImageViz
+from frcnn.processing_image import Preprocess
+from frcnn.modeling_frcnn import GeneralizedRCNN
+from frcnn.utils import Config
 max_detections = 36
 frcnn_config = json.load(open("frcnn/config.jsonl"))
 frcnn_config = Config(frcnn_config)
-frcnn = GeneralizedRCNN(frcnn_config).half().cuda()
-image_preprocessor = Preprocess(frcnn_config).half().cuda()
-
-img_url = 'image2.png' 
+image_preprocessor= Preprocess(frcnn_config).half().cuda()
+box_segmentation_model= GeneralizedRCNN.from_pretrained("unc-nlp/frcnn-vg-finetuned", frcnn_config).half().cuda()
+    
+img_url = 'image.png' 
 raw_image = Image.open(img_url).convert('RGB')
 frcnn_output = decode_image(asarray(raw_image),  frcnn, image_preprocessor, max_detections=max_detections)