From 677317b81a5aa96db8c4837ba56078e2014091e3 Mon Sep 17 00:00:00 2001
From: Jobil Thomas <63110554+Jobilthomas@users.noreply.github.com>
Date: Tue, 20 Feb 2024 17:45:24 +0530
Subject: [PATCH] Update voc.py

fix bug in _write_voc_results_file.
---
 yolox/data/datasets/voc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yolox/data/datasets/voc.py b/yolox/data/datasets/voc.py
index bdacd8019..cae2c3eba 100644
--- a/yolox/data/datasets/voc.py
+++ b/yolox/data/datasets/voc.py
@@ -261,7 +261,7 @@ def _write_voc_results_file(self, all_boxes):
                 for im_ind, index in enumerate(self.ids):
                     index = index[1]
                     dets = all_boxes[cls_ind][im_ind]
-                    if dets == []:
+                    if dets.shape[0] == 0:
                         continue
                     for k in range(dets.shape[0]):
                         f.write(