Update README.md (#82)

Correction of bounding box format on line 73 and line 118. Its (x1, y1, x2, y2) instead of (x1, x2, y1, y2)
fcakyon · Jan 14, 2022 · 9866404 · 9866404
1 parent f8ec480
commit 9866404
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -70,7 +70,7 @@ results = model(img, augment=True)
 
 # parse results
 predictions = results.pred[0]
-boxes = predictions[:, :4] # x1, x2, y1, y2
+boxes = predictions[:, :4] # x1, y1, x2, y2
 scores = predictions[:, 4]
 categories = predictions[:, 5]
 
@@ -115,7 +115,7 @@ results = yolov5.predict([image1, image2], size=1280, augment=True)
 
 # parse results
 predictions = results.pred[0]
-boxes = predictions[:, :4] # x1, x2, y1, y2
+boxes = predictions[:, :4] # x1, y1, x2, y2
 scores = predictions[:, 4]
 categories = predictions[:, 5]