Move tests outside package (#104)
This commit is contained in:
parent
8028e2b1b8
commit
8b6466f731
7 changed files with 0 additions and 0 deletions
32
tests/functional/test_loaders.py
Normal file
32
tests/functional/test_loaders.py
Normal 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()
|
||||
Loading…
Add table
Add a link
Reference in a new issue