From 0af06dd6693ea0780fdf7bfc3579cb21230a15b9 Mon Sep 17 00:00:00 2001 From: "hongliang.yuan" Date: Tue, 22 Apr 2025 09:27:41 +0800 Subject: [PATCH] sync ssd code --- .../ssd/tensorflow/dataset/convert_voc_sample_tfrecords.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv/detection/ssd/tensorflow/dataset/convert_voc_sample_tfrecords.py b/cv/detection/ssd/tensorflow/dataset/convert_voc_sample_tfrecords.py index 6fe35a4a..8231e7a3 100644 --- a/cv/detection/ssd/tensorflow/dataset/convert_voc_sample_tfrecords.py +++ b/cv/detection/ssd/tensorflow/dataset/convert_voc_sample_tfrecords.py @@ -326,7 +326,7 @@ def _process_image_files(name, directory, all_records, num_shards): num_shards: integer number of shards for this data set. """ # Break all images into batches with a [ranges[i][0], ranges[i][1]]. - spacing = np.linspace(0, len(all_records), FLAGS.num_threads + 1).astype(np.int) + spacing = np.linspace(0, len(all_records), FLAGS.num_threads + 1).astype(np.int32) ranges = [] threads = [] for i in range(len(spacing) - 1): -- Gitee