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
username: stormsson
web@ultralytics.com:
avatar: null
username: null
avatar: https://avatars.githubusercontent.com/u/135830346?v=4
username: UltralyticsAssistant
xinwang614@gmail.com:
avatar: https://avatars.githubusercontent.com/u/17264618?v=4
username: GreatV

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

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

Loading…
Cancel
Save