tarasevicius commited on
Commit
887ab5c
·
1 Parent(s): 62f55ce

feat: set line thickness

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ model_path = hf_hub_download(repo_id='ethandavey/yoloV5-coursework-model', filen
9
 
10
  # Load the YOLOv5 model
11
  model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, source='github')
 
 
12
 
13
  # Print model class names
14
  print("Model class names:", model.names)
 
9
 
10
  # Load the YOLOv5 model
11
  model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, source='github')
12
+ model.line_thickness = 1 # Set the thickness of the bounding box lines
13
+
14
 
15
  # Print model class names
16
  print("Model class names:", model.names)