Add YOLOv7 support

2022-08-12 16:33:26 -03:00
parent 23547b19b2
commit 80d08990a0
10 changed files with 586 additions and 47 deletions
--- a/utils/gen_wts_ppyoloe.py
+++ b/utils/gen_wts_ppyoloe.py
@@ -333,7 +333,6 @@ def export_model():

    merge_config(FLAGS.opt)
    check_config(cfg)
-    check_gpu(cfg.use_gpu)
    check_version()

    trainer = Trainer(cfg, mode='test')