homebrew-core/Formula/caffe.rb

67 lines
2 KiB
Ruby
Raw Normal View History

class Caffe < Formula
desc "Fast open framework for deep learning"
homepage "https://caffe.berkeleyvision.org/"
url "https://github.com/BVLC/caffe/archive/1.0.tar.gz"
sha256 "71d3c9eb8a183150f965a465824d01fe82826c22505f7aa314f700ace03fa77f"
2018-09-09 21:18:34 +00:00
revision 4
2018-03-01 16:25:40 +00:00
bottle do
2018-09-12 00:03:01 +00:00
sha256 "0bdb0a30abe7fa59b042452fb40d71915adb008d06fd6440eddebbeda002f32f" => :high_sierra
sha256 "b95d74d2bcc906afdd717a5ecd1c79ada8c6acb89215a2f8928086882a3ad625" => :sierra
sha256 "2d6c0ba3d484c41e76eb821048e0c0c97dcd851d236f0e86d86a9afeff9cfffd" => :el_capitan
2018-03-01 16:25:40 +00:00
end
depends_on "cmake" => :build
depends_on "boost"
depends_on "gflags"
depends_on "glog"
depends_on "hdf5"
depends_on "protobuf"
depends_on "szip"
depends_on "leveldb" => :optional
depends_on "lmdb" => :optional
depends_on "opencv" => :optional
depends_on "snappy" if build.with?("leveldb")
resource "test_model_weights" do
url "http://dl.caffe.berkeleyvision.org/bvlc_reference_caffenet.caffemodel"
sha256 "472d4a06035497b180636d8a82667129960371375bd10fcb6df5c6c7631f25e0"
end
needs :cxx11
def install
ENV.cxx11
args = std_cmake_args + %w[
-DCPU_ONLY=ON
-DUSE_NCCL=OFF
-DBUILD_SHARED_LIBS=ON
-DBUILD_python=OFF
-DBUILD_matlab=OFF
-DBUILD_docs=OFF
-DBUILD_python_layer=OFF
-DALLOW_LMDB_NOLOCK=OFF
-DUSE_OPENMP=OFF
]
args << "-DUSE_OPENCV=" + (build.with?("opencv") ? "ON" : "OFF")
args << "-DUSE_LMDB=" + (build.with?("lmdb") ? "ON" : "OFF")
args << "-DUSE_LEVELDB=" + (build.with?("leveldb") ? "ON" : "OFF")
system "cmake", ".", *args
system "make", "install"
pkgshare.install "models"
end
test do
model = "bvlc_reference_caffenet"
m_path = "#{pkgshare}/models/#{model}"
resource("test_model_weights").stage do
system "#{bin}/caffe", "test",
"-model", "#{m_path}/deploy.prototxt",
"-solver", "#{m_path}/solver.prototxt",
"-weights", "#{model}.caffemodel"
end
end
end