From 56100f24141db14459882b51bd7ea210e1970895 Mon Sep 17 00:00:00 2001 From: JiaQi Xu <47347516+bubbliiiing@users.noreply.github.com> Date: Fri, 29 May 2020 12:20:41 +0800 Subject: [PATCH] Update test.py --- test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index b59de15..a176f1b 100644 --- a/test.py +++ b/test.py @@ -1,8 +1,8 @@ -from nets.yolo3 import yolo_body +from nets.yolo4 import yolo_body from keras.layers import Input inputs = Input([416,416,3]) model = yolo_body(inputs,3,80) model.summary() for i,layer in enumerate(model.layers): - print(i,layer.name) \ No newline at end of file + print(i,layer.name) -- GitLab