Skip to content

Commit bc8f7b2

Browse files
author
Zuber Khan
committed
Update
1 parent b01e7d0 commit bc8f7b2

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

+helper/detectYolov9.m

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@
4747
out = cat(1, out1r, out2r, out3r);
4848

4949
% Post-processing ONNX Layers implementation in MATLAB
50-
% This is specific to Yolov9c model
5150
nclasses = 80;
5251
dim1 = size(out, 1);
5352
dim2 = size(out, 2);
@@ -58,7 +57,6 @@
5857
splitout2 = out(:, endIndex+1:dim2);
5958
splitout2 = sigmoid(splitout2);
6059

61-
% Hard-coded values for 144 channels
6260
splitout1 = reshape(splitout1, [dim1, 16, 4]);
6361
splitout1 = permute(splitout1, [1, 3, 2]);
6462
splitout1 = permute(splitout1, [2, 1, 3]);
@@ -102,7 +100,6 @@
102100

103101
% Get the final feature map
104102
featureMap = permute(combout, [2, 1]);
105-
%finalout = reshape(final, [1, size(final, 1), size(final, 2)]);
106103

107104
% Post-process obtained feature map
108105
output = helper.postprocess(featureMap,imgp, image);

0 commit comments

Comments
 (0)