質問編集履歴
4
追記の追加
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -215,3 +215,117 @@
|
|
|
215
215
|
break
|
|
216
216
|
|
|
217
217
|
```
|
|
218
|
+
|
|
219
|
+
|
|
220
|
+
|
|
221
|
+
|
|
222
|
+
|
|
223
|
+
追記
|
|
224
|
+
|
|
225
|
+
```Python
|
|
226
|
+
|
|
227
|
+
def main():
|
|
228
|
+
|
|
229
|
+
|
|
230
|
+
|
|
231
|
+
img = get_image()
|
|
232
|
+
|
|
233
|
+
|
|
234
|
+
|
|
235
|
+
# リサイズ
|
|
236
|
+
|
|
237
|
+
image_data = preprocess(img)
|
|
238
|
+
|
|
239
|
+
image_size = np.array([img.size[1], img.size[0]], dtype=np.int32).reshape(1, 2)
|
|
240
|
+
|
|
241
|
+
|
|
242
|
+
|
|
243
|
+
# 学習済みモデルからセッションを作る
|
|
244
|
+
|
|
245
|
+
session = onnxruntime.InferenceSession('yolov5s.onnx')
|
|
246
|
+
|
|
247
|
+
|
|
248
|
+
|
|
249
|
+
# #元の画像の表示
|
|
250
|
+
|
|
251
|
+
# plt.imshow(img)
|
|
252
|
+
|
|
253
|
+
# plt.show()
|
|
254
|
+
|
|
255
|
+
|
|
256
|
+
|
|
257
|
+
#input output 名前取得
|
|
258
|
+
|
|
259
|
+
print("\n")
|
|
260
|
+
|
|
261
|
+
print("input:")
|
|
262
|
+
|
|
263
|
+
for session_input in session.get_inputs():
|
|
264
|
+
|
|
265
|
+
input_name = session_input.name
|
|
266
|
+
|
|
267
|
+
input_name_img_shape = session_input.shape
|
|
268
|
+
|
|
269
|
+
print(session_input.name, session_input.shape, session_input.type)
|
|
270
|
+
|
|
271
|
+
|
|
272
|
+
|
|
273
|
+
|
|
274
|
+
|
|
275
|
+
print("\n")
|
|
276
|
+
|
|
277
|
+
# 出力のラベル名の確認
|
|
278
|
+
|
|
279
|
+
print("output:")
|
|
280
|
+
|
|
281
|
+
for session_output in session.get_outputs():
|
|
282
|
+
|
|
283
|
+
print(session_output.name, session_output.shape, session_output.type)
|
|
284
|
+
|
|
285
|
+
```
|
|
286
|
+
|
|
287
|
+
を実行したところ，
|
|
288
|
+
|
|
289
|
+
|
|
290
|
+
|
|
291
|
+
```terminal
|
|
292
|
+
|
|
293
|
+
|
|
294
|
+
|
|
295
|
+
input:
|
|
296
|
+
|
|
297
|
+
images [1, 3, 640, 640] tensor(float)
|
|
298
|
+
|
|
299
|
+
|
|
300
|
+
|
|
301
|
+
|
|
302
|
+
|
|
303
|
+
output:
|
|
304
|
+
|
|
305
|
+
output [1, 25200, 85] tensor(float)
|
|
306
|
+
|
|
307
|
+
351 [1, 3, 80, 80, 85] tensor(float)
|
|
308
|
+
|
|
309
|
+
420 [1, 3, 40, 40, 85] tensor(float)
|
|
310
|
+
|
|
311
|
+
489 [1, 3, 20, 20, 85] tensor(float)
|
|
312
|
+
|
|
313
|
+
```
|
|
314
|
+
|
|
315
|
+
と出ました．
|
|
316
|
+
|
|
317
|
+
|
|
318
|
+
|
|
319
|
+
ここから
|
|
320
|
+
|
|
321
|
+
```
|
|
322
|
+
|
|
323
|
+
outputs_index = session.run([output_name_boxes, output_name_scores, output_name_indices],
|
|
324
|
+
|
|
325
|
+
{input_name: image_data, input_name_img_shape: image_size})
|
|
326
|
+
|
|
327
|
+
```
|
|
328
|
+
|
|
329
|
+
|
|
330
|
+
|
|
331
|
+
に落とし込みたいのですが，どうすればよいか分かりません．
|
3
タグの追加
test
CHANGED
|
File without changes
|
test
CHANGED
|
File without changes
|
2
実現したいことの追記
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -12,6 +12,10 @@
|
|
|
12
12
|
|
|
13
13
|
|
|
14
14
|
|
|
15
|
+
PyTorchは使えないので，YOLOv5の元のコードをコピペすることはできません．
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
|
|
15
19
|
|
|
16
20
|
|
|
17
21
|
### 該当のソースコード
|
1
タイトルの変更
test
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
YOLOv5のバウンディングボックスの処理をPythonで記述したい
|
|
1
|
+
ONNXに変換したYOLOv5のバウンディングボックスの処理をPythonで記述したい
|
test
CHANGED
|
File without changes
|