Move tests outside package (#104)

This commit is contained in:
Ayush Chaurasia
2022-12-27 22:16:07 +05:30
committed by GitHub
parent 8028e2b1b8
commit 8b6466f731
7 changed files with 0 additions and 0 deletions

View File

@ -0,0 +1,32 @@
from ultralytics.nn.tasks import DetectionModel
from ultralytics.yolo.utils.checks import check_yaml
def test_model_parser():
cfg = check_yaml("../assets/dummy_model.yaml") # check YAML
# Create model
model = DetectionModel(cfg)
print(model)
'''
# Options
if opt.line_profile: # profile layer by layer
model(im, profile=True)
elif opt.profile: # profile forward-backward
results = profile(input=im, ops=[model], n=3)
elif opt.test: # test all models
for cfg in Path(ROOT / 'models').rglob('yolo*.yaml'):
try:
_ = Model(cfg)
except Exception as e:
print(f'Error in {cfg}: {e}')
else: # report fused model summary
model.fuse()
'''
if __name__ == "__main__":
test_model_parser()