Merge branch 'main' into softnms

softnms
Burhan 4 months ago
commit 8fa5cc184a
  1. 4
      docs/mkdocs_github_authors.yaml
  2. 3
      tests/test_python.py
  3. 6
      ultralytics/cfg/datasets/Objects365.yaml

@ -116,8 +116,8 @@ stormsson@users.noreply.github.com:
avatar: https://avatars.githubusercontent.com/u/1133032?v=4 avatar: https://avatars.githubusercontent.com/u/1133032?v=4
username: stormsson username: stormsson
web@ultralytics.com: web@ultralytics.com:
avatar: null avatar: https://avatars.githubusercontent.com/u/135830346?v=4
username: null username: UltralyticsAssistant
xinwang614@gmail.com: xinwang614@gmail.com:
avatar: https://avatars.githubusercontent.com/u/17264618?v=4 avatar: https://avatars.githubusercontent.com/u/17264618?v=4
username: GreatV username: GreatV

@ -196,13 +196,14 @@ def test_all_model_yamls():
YOLO(m.name) YOLO(m.name)
@pytest.mark.skipif(WINDOWS, reason="Windows slow CI export bug https://github.com/ultralytics/ultralytics/pull/16003")
def test_workflow(): def test_workflow():
"""Test the complete workflow including training, validation, prediction, and exporting.""" """Test the complete workflow including training, validation, prediction, and exporting."""
model = YOLO(MODEL) model = YOLO(MODEL)
model.train(data="coco8.yaml", epochs=1, imgsz=32, optimizer="SGD") model.train(data="coco8.yaml", epochs=1, imgsz=32, optimizer="SGD")
model.val(imgsz=32) model.val(imgsz=32)
model.predict(SOURCE, imgsz=32) model.predict(SOURCE, imgsz=32)
model.export(format="torchscript") model.export(format="torchscript") # WARNING: Windows slow CI export bug
def test_predict_callback_and_setup(): def test_predict_callback_and_setup():

@ -113,7 +113,7 @@ names:
95: Pot 95: Pot
96: Cow 96: Cow
97: Cake 97: Cake
98: Dinning Table 98: Dining Table
99: Sheep 99: Sheep
100: Hanger 100: Hanger
101: Blackboard/Whiteboard 101: Blackboard/Whiteboard
@ -304,7 +304,7 @@ names:
286: Hammer 286: Hammer
287: Cue 287: Cue
288: Avocado 288: Avocado
289: Hamimelon 289: Hami melon
290: Flask 290: Flask
291: Mushroom 291: Mushroom
292: Screwdriver 292: Screwdriver
@ -328,7 +328,7 @@ names:
310: Dishwasher 310: Dishwasher
311: Crab 311: Crab
312: Hoverboard 312: Hoverboard
313: Meat ball 313: Meatball
314: Rice Cooker 314: Rice Cooker
315: Tuba 315: Tuba
316: Calculator 316: Calculator

Loading…
Cancel
Save