apollo::perception::inference::BatchStream类 参考

#include <batch_stream.h>

apollo::perception::inference::BatchStream 的协作图:

Public 成员函数
	BatchStream (int batchSize, int maxBatches, std::string data_path)
	BatchStream ()
void	reset (int firstBatch)
bool	next ()
void	skip (int skipCount)
float *	getBatch ()
int	getBatchesRead () const
int	getBatchSize () const
nvinfer1::DimsNCHW	getDims () const

详细描述

在文件 batch_stream.h 第 36 行定义.

构造及析构函数说明

BatchStream() [1/2]

apollo::perception::inference::BatchStream::BatchStream	(	int	batchSize,
		int	maxBatches,
		std::string	data_path
	)

在文件 batch_stream.cc 第 35 行定义.

    : mBatchSize(batchSize), mMaxBatches(maxBatches), mPath(dataPath) {
  FILE *file = fopen((mPath + "Batch0").c_str(), "rb");
  if (file != nullptr) {
    int d[4];
    int fs = static_cast<int>(fread(d, sizeof(int), 4, file));
    CHECK_EQ(fs, 4);
    mDims = nvinfer1::DimsNCHW{d[0], d[1], d[2], d[3]};
    fclose(file);
    mImageSize = mDims.c() * mDims.h() * mDims.w();
    mBatch.resize(mBatchSize * mImageSize, 0);
    mFileBatch.resize(mDims.n() * mImageSize, 0);
    reset(0);
  }
}

BatchStream() [2/2]

apollo::perception::inference::BatchStream::BatchStream

(

)

在文件 batch_stream.cc 第 51 行定义.

: mPath("") {}

成员函数说明

getBatch()

float * apollo::perception::inference::BatchStream::getBatch ( )

inline

在文件 batch_stream.h 第 50 行定义.

{ return &mBatch[0]; }

getBatchesRead()

int apollo::perception::inference::BatchStream::getBatchesRead ( ) const

inline

在文件 batch_stream.h 第 51 行定义.

{ return mBatchCount; }

getBatchSize()

int apollo::perception::inference::BatchStream::getBatchSize ( ) const

inline

在文件 batch_stream.h 第 52 行定义.

{ return mBatchSize; }

getDims()

nvinfer1::DimsNCHW apollo::perception::inference::BatchStream::getDims ( ) const

inline

在文件 batch_stream.h 第 53 行定义.

{ return mDims; }

next()

bool apollo::perception::inference::BatchStream::next

(

)

在文件 batch_stream.cc 第 62 行定义.

                       {
  if (mBatchCount == mMaxBatches) {
    return false;
  }
 
  for (int csize = 1, batchPos = 0; batchPos < mBatchSize;
       batchPos += csize, mFileBatchPos += csize) {
    CHECK_GT(mFileBatchPos, 0);
    CHECK_LE(mFileBatchPos, mDims.n());
    // mMaxBatches > number of batches in the files
    if (mFileBatchPos == mDims.n() && !update()) {
      return false;
    }
 
    // copy the smaller of: elements left to fulfill the request,
    // or elements left in the file buffer.
    csize = std::min(mBatchSize - batchPos, mDims.n() - mFileBatchPos);
    std::copy_n(getFileBatch() + mFileBatchPos * mImageSize, csize * mImageSize,
                getBatch() + batchPos * mImageSize);
  }
  mBatchCount++;
  return true;
}

reset()

void apollo::perception::inference::BatchStream::reset

(

int

firstBatch

)

在文件 batch_stream.cc 第 53 行定义.

                                      {
  if (mPath != "") {
    mBatchCount = 0;
    mFileCount = 0;
    mFileBatchPos = mDims.n();
    skip(firstBatch);
  }
}

skip()

void apollo::perception::inference::BatchStream::skip

(

int

skipCount

)

在文件 batch_stream.cc 第 86 行定义.

                                    {
  if (mBatchSize >= mDims.n() && mBatchSize % mDims.n() == 0 &&
      mFileBatchPos == mDims.n()) {
    mFileCount += skipCount * mBatchSize / mDims.n();
    return;
  }
 
  int x = mBatchCount;
  for (int i = 0; i < skipCount; ++i) {
    next();
  }
  mBatchCount = x;
}

---

该类的文档由以下文件生成:

• modules/perception/common/inference/tensorrt/batch_stream.h
• modules/perception/common/inference/tensorrt/batch_stream.cc