diff --git a/DESCRIPTION b/DESCRIPTION index 025ba44..f465539 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -1,6 +1,6 @@ Package: ovmlpy Title: Machine Learning Tools for Volleyball -Version: 0.1.4 +Version: 0.1.5 Authors@R: c(person("Ben", "Raymond", role = c("aut", "cre"), email = "ben@untan.gl"), person("Adrien", "Ickowicz", role = "aut"), person("openvolley.org", role = "org")) diff --git a/R/yolov7_python.R b/R/yolov7_python.R index 3c37422..9376ee6 100644 --- a/R/yolov7_python.R +++ b/R/yolov7_python.R @@ -12,7 +12,7 @@ ovml_yolo7_python_setup <- function() { ## 2. create the yolov7 virtual environment if needed, or find the existing on envname <- ovml_yolo7_python_envname() if (!envname %in% reticulate::virtualenv_list()) { - reticulate::virtualenv_create(envname, python = reticulate::install_python(), packages = c("opencv-python", "torch", "pandas", "torchvision", "tqdm", "matplotlib", "seaborn", "pyyaml")) + reticulate::virtualenv_create(envname, python = reticulate::install_python(), packages = c("opencv-python", "torch", "pandas", "torchvision", "tqdm", "matplotlib", "seaborn", "pyyaml", "scipy")) } ## 3. install yolov7 if needed from https://github.com/WongKinYiu/yolov7 @@ -30,7 +30,7 @@ ovml_yolo7_python_envname <- function() "ovml-yolov7" #' @rdname ovml_yolo7_python_setup #' @export -ovml_yolo7_python_envpath <- function() file.path(reticulate::virtualenv_root(), ovml_yolo7_python_envname) +ovml_yolo7_python_envpath <- function() file.path(reticulate::virtualenv_root(), ovml_yolo7_python_envname()) ovml_yolo7_python_dir <- function(install = FALSE) { y7dir <- rappdirs::user_data_dir("ovml", appauthor = "openvolley")