add model weights

README.md

@@ -50,6 +50,49 @@ This is to show how to create a computer vision model for applications such as a
 
 ### Direct Use
 
+- Install [yolov5](https://github.com/fcakyon/yolov5-pip):
+
+```bash
+pip install -U yolov5
+```
+
+- Load model and perform prediction:
+
+```python
+import yolov5
+
+# load model
+model = yolov5.load('ljamesdatascience/BillboardAdvertDetectionYOLOV5')
+  
+# set model parameters
+model.conf = 0.25  # NMS confidence threshold
+model.iou = 0.45  # NMS IoU threshold
+model.agnostic = False  # NMS class-agnostic
+model.multi_label = False  # NMS multiple labels per box
+model.max_det = 1000  # maximum number of detections per image
+
+# set image
+img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
+
+# perform inference
+results = model(img, size=640)
+
+# inference with test time augmentation
+results = model(img, augment=True)
+
+# parse results
+predictions = results.pred[0]
+boxes = predictions[:, :4] # x1, y1, x2, y2
+scores = predictions[:, 4]
+categories = predictions[:, 5]
+
+# show detection bounding boxes on image
+results.show()
+
+# save results into "results/" folder
+results.save(save_dir='results/')
+```
+
 
 ### Results 
 | Class | Images | Instances | P     | R     | mAP50 |

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14b36c2d26ce19778db6cfb6d1b88a74043c1eea106e2be1ba0897daac458d69
+size 14401448