thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

DeepStream 6.1 update

Browse Source
This commit is contained in:
Marcos Luciano
2022-05-26 08:05:06 -03:00
parent 4192a16751
commit ed387abf5d
3 changed files with 259 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nvdsinfer_custom_impl_Yolo/yoloPlugins.cpp
View File

@@ -193,8 +193,8 @@ YoloLayer::configureWithFormat (
assert(inputDims != nullptr);
}
int YoloLayer::enqueue (
int batchSize, void const* const* inputs, void* const* outputs, void* workspace,
int32_t YoloLayer::enqueue (
int32_t batchSize, void const* const* inputs, void* const* outputs, void* workspace,
cudaStream_t stream) noexcept
{
if (m_Type == 2) { // YOLOR incorrect param: scale_x_y = 2.0

4
nvdsinfer_custom_impl_Yolo/yoloPlugins.h
View File

@@ -81,8 +81,8 @@ public:
int initialize () noexcept override { return 0; }
void terminate () noexcept override {}
size_t getWorkspaceSize (int maxBatchSize) const noexcept override { return 0; }
int enqueue (
int batchSize, void const* const* inputs, void* const* outputs,
int32_t enqueue (
int32_t batchSize, void const* const* inputs, void* const* outputs,
void* workspace, cudaStream_t stream) noexcept override;
size_t getSerializationSize() const noexcept override;
void serialize (void* buffer) const noexcept override;