henry000 commited on
Commit
89b0ef8
Β·
1 Parent(s): 16c650d

πŸ“ [Update] pip+git install link

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. yolo/tools/drawer.py +2 -2
README.md CHANGED
@@ -20,7 +20,7 @@ Welcome to the official implementation of YOLOv7 and YOLOv9. This repository wil
20
  - This is the official YOLO model implementation with an MIT License.
21
  - For quick deployment: you can directly install by pip+git:
22
  ```shell
23
- pip install git+git@github.com:WongKinYiu/YOLO.git
24
  yolo task.data.source=0 # source could be a single file, video, image folder, webcam ID
25
  ```
26
 
 
20
  - This is the official YOLO model implementation with an MIT License.
21
  - For quick deployment: you can directly install by pip+git:
22
  ```shell
23
+ pip install git+https://github.com/WongKinYiu/YOLO.git
24
  yolo task.data.source=0 # source could be a single file, video, image folder, webcam ID
25
  ```
26
 
yolo/tools/drawer.py CHANGED
@@ -37,9 +37,9 @@ def draw_bboxes(
37
  draw = ImageDraw.Draw(img, "RGBA")
38
 
39
  try:
40
- font = ImageFont.truetype("arial.ttf", label_size)
41
  except IOError:
42
- font = ImageFont.load_default(label_size)
43
 
44
  for bbox in bboxes:
45
  class_id, x_min, y_min, x_max, y_max, *conf = [float(val) for val in bbox]
 
37
  draw = ImageDraw.Draw(img, "RGBA")
38
 
39
  try:
40
+ font = ImageFont.truetype("arial.ttf", int(label_size))
41
  except IOError:
42
+ font = ImageFont.load_default(int(label_size))
43
 
44
  for bbox in bboxes:
45
  class_id, x_min, y_min, x_max, y_max, *conf = [float(val) for val in bbox]