Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
caffe-pspnet-gpu-dev / usr / include / caffe / util / matio_io.hpp
Size: Mime:
#ifndef CAFFE_UTIL_MATIO_IO_H_
#define CAFFE_UTIL_MATIO_IO_H_

#include <unistd.h>
#include <string>

#include "google/protobuf/message.h"
#include "caffe/blob.hpp"
#include "caffe/common.hpp"
#include "caffe/proto/caffe.pb.h"

namespace caffe {

using ::google::protobuf::Message;

template <typename Dtype>
void ReadBlobFromMat(const char *fname, Blob<Dtype>* blob);

template <typename Dtype>
void WriteBlobToMat(const char *fname, bool write_diff,
   Blob<Dtype>* blob);

}  // namespace caffe

#endif   // CAFFE_UTIL_IO_H_