R/tensorflow.R

In mikeyEcology/MLWIC2: Machine Learning for Wildlife Image Classification

#' Install TensorFlow for use with \code{MLWIC2}
#'
#' \code{MLWIC2} requires an installation of tensorflow that can be used by Python.
#'  You need to use this before using \code{classify} or \code{train}. If this is your first time using
#'  this function, you should see additional documentation at https://github.com/mikeyEcology/MLWIC2.
#'  This function will install tensorflow on Linux machines; if you are using Windows,
#'  you will need to install tensorflow on your own following the directions here:
#'  https://www.tensorflow.org/install/install_windows. I recommend using the installation with
#'  Anaconda. Install tensorflow version 1.14. Version 2.0 or greater does not function
#'  properly.
#'
#' @param os The operating system on your computer. Options are "Mac" or "Ubuntu".
#'  Specifying "Windows" will thrown an error because we cannot automatically install
#'  TensorFlow on Windows at this time.
#' @export

tensorflow <- function(os="Mac"){
  
  ## Check for python
  vpython <- system("pyv=\"$(python -V 2>&1)\" | echo $pyv | grep \"Python\"") ## come back to this
  
  if(vpython == TRUE){
    print("Python is installed. Installing homebrew, protobuf, pip, and tensorflow.\n")
    
    if(os == "Mac"){
      print("Installing some software. You might need to enter your password to allow install.\n")
      system("/usr/bin/ruby -e \"$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)\"")
      system("brew install protobuf")
      system("sudo easy_install --upgrade pip")
      system("sudo easy_install --upgrade six")
      system("sudo conda update --all")
      system("sudo pip install tensorflow==1.14")
      ## Something to validate installation, beyond this.
      #system("python import_tf.py")
      
      # I think I need to add: conda install tensorflow
      
    }else if(os == "Ubuntu"){
      print("Installing some software. You might need to enter your password to allow install.\n")
      system("sudo apt-get install python-pip python-dev")   # for Python 2.7
      system("pip install tensorflow==1.14")
      
      #system("python import_tf.py")
      
    }else if(os == "Windows"){
      system("pip install --upgrade pip")
      system("pip install tensorflow==1.14")
      print("Installing tensorflow on Windows is more complicated and might not work directly from R. If you have problems see https://www.tensorflow.org/install/install_windows for tensorflow installation instructions. Install tensorflow version 1.14")
      
    }else{
      print('Specify operating system - \"Mac\", \"Windows\", or \"Ubuntu\"')
    }
    
  }else{
    print("Python needs to be installed.")
  }
  
}