🔥 [Remove] main function in yolo.py
Browse files- yolo/model/yolo.py +0 -6
yolo/model/yolo.py
CHANGED
|
@@ -102,9 +102,3 @@ def get_model(model_cfg: dict) -> YOLO:
|
|
| 102 |
model = YOLO(model_cfg)
|
| 103 |
logger.info("✅ Success load model")
|
| 104 |
return model
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
if __name__ == "__main__":
|
| 108 |
-
model_cfg = load_model_cfg("v7-base")
|
| 109 |
-
|
| 110 |
-
YOLO(model_cfg)
|
|
|
|
| 102 |
model = YOLO(model_cfg)
|
| 103 |
logger.info("✅ Success load model")
|
| 104 |
return model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|