You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
332 B
14 lines
332 B
2 years ago
|
from ultralytics.yolo import YOLO
|
||
|
|
|
||
|
|
|
||
|
|
def test_model():
|
||
|
|
model = YOLO()
|
||
|
|
model.new("assets/dummy_model.yaml")
|
||
|
|
model.model = "squeezenet1_0" # temp solution before get_model is implemented
|
||
|
|
# model.load("yolov5n.pt")
|
||
|
|
model.train(data="imagenette160", epochs=1, lr0=0.01)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_model()
|