Advertisement
SirMeme

SIR MEME'S HELP SHEET 4 AIMMY GUIDE

Apr 2nd, 2024
393
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.71 KB | None | 0 0
  1.  
  2. THIS IS THE COMMAND YOU EXECUTE TO START TRAINING THE MODEL, READ NOTE BELOW TO SEE INFO ON EPOCHS:
  3. yolo task=detect mode=train imgsz=640 data=data.yaml epochs=100 batch=16 name=putyourmodelnamehere
  4.  
  5.  
  6.  
  7. USE THIS AFTER YOU'VE FINISHED TRAINING TO CONVERT YOUR MODEL TO ONNX SO YOU CAN USE IT WITH AIMMY:
  8. yolo export model=best.pt format=onnx
  9.  
  10.  
  11.  
  12.  
  13.  
  14. NOTE:
  15. use 1 - 1000 epochs (the more epochs the longer the training but better end results for model)
  16.  
  17.  
  18.  
  19.  
  20.  
  21.  
  22. YOU NEED THIS:
  23. pip install ultralytics
  24.  
  25. YOU ONLY NEED THIS IF THE ABOVE IS NOT WORKING (get requirements.txt from:https://docs.ultralytics.com/quickstart/)
  26. pip install -r requirements.txt
  27.  
  28.  
  29. YOU NEED THIS:
  30. https://pytorch.org/get-started/locally/
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement