ovml_yolo: Construct YOLO network

In openvolley/ovml: Machine Learning Tools for Volleyball

Construct YOLO network

Description

Construct YOLO network

Usage

ovml_yolo(version = 4, device = "cuda", weights_file = "auto", class_labels)

Arguments

version	integer or string: one of 3 : YOLO v3 4 : YOLO v4 "4-tiny" : YOLO v4-tiny "4-mvb" : an experimental network trained specifically to detect (only) volleyballs "4-tiny-mvb" : the v4-tiny version of the same 7 or "7-tiny" : YOLO v7 or v7-tiny
device	string: "cpu" or "cuda"
weights_file	string: either the path to the weights file that already exists on your system or "auto". If "auto", the weights file will be downloaded if necessary and stored in the directory given by ovml_cache_dir()
class_labels	character: the class labels used for network training. If missing or NULL, these default to ovml_class_labels("coco") for all models except "mvb" models, which use ovml_class_labels("mvb")

Value

A YOLO network object

References

https://github.com/pjreddie/darknet, https://github.com/WongKinYiu/yolov7

Examples

## Not run: 
  dn <- ovml_yolo()
  img <- ovml_example_image()
  res <- ovml_yolo_detect(dn, img)
  ovml_ggplot(img, res)

## End(Not run)

openvolley/ovml documentation built on Aug. 2, 2022, 11:53 a.m.