mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 22:46:35 +00:00
Update configuration and test structure for improved clarity and functionality
- Modified `.gitignore` to include test fixture data while excluding test results. - Updated `config.yaml` to change the model from 'yolo11m.yaml' to 'yolo26m.pt'. - Enhanced `.cursor/rules/coderule.mdc` with additional guidelines for test environment consistency and infrastructure handling. - Revised autopilot state management in `_docs/_autopilot_state.md` to reflect current progress and tasks. - Removed outdated augmentation tests and adjusted dataset formation tests to align with the new structure. These changes streamline the configuration and testing processes, ensuring better organization and clarity in the project.
This commit is contained in:
@@ -55,15 +55,15 @@ def test_bt_dsf_01_split_ratio_70_20_10(
|
||||
constants_patch,
|
||||
fixture_images_dir,
|
||||
fixture_labels_dir,
|
||||
100,
|
||||
20,
|
||||
set(),
|
||||
)
|
||||
# Act
|
||||
train.form_dataset()
|
||||
# Assert
|
||||
assert _count_jpg(Path(today_ds, "train", "images")) == 70
|
||||
assert _count_jpg(Path(today_ds, "valid", "images")) == 20
|
||||
assert _count_jpg(Path(today_ds, "test", "images")) == 10
|
||||
assert _count_jpg(Path(today_ds, "train", "images")) == 14
|
||||
assert _count_jpg(Path(today_ds, "valid", "images")) == 4
|
||||
assert _count_jpg(Path(today_ds, "test", "images")) == 2
|
||||
|
||||
|
||||
def test_bt_dsf_02_six_subdirectories(
|
||||
@@ -80,7 +80,7 @@ def test_bt_dsf_02_six_subdirectories(
|
||||
constants_patch,
|
||||
fixture_images_dir,
|
||||
fixture_labels_dir,
|
||||
100,
|
||||
20,
|
||||
set(),
|
||||
)
|
||||
# Act
|
||||
@@ -95,7 +95,7 @@ def test_bt_dsf_02_six_subdirectories(
|
||||
assert (base / "test" / "labels").is_dir()
|
||||
|
||||
|
||||
def test_bt_dsf_03_total_files_one_hundred(
|
||||
def test_bt_dsf_03_total_files_twenty(
|
||||
monkeypatch,
|
||||
tmp_path,
|
||||
constants_patch,
|
||||
@@ -109,7 +109,7 @@ def test_bt_dsf_03_total_files_one_hundred(
|
||||
constants_patch,
|
||||
fixture_images_dir,
|
||||
fixture_labels_dir,
|
||||
100,
|
||||
20,
|
||||
set(),
|
||||
)
|
||||
# Act
|
||||
@@ -120,7 +120,7 @@ def test_bt_dsf_03_total_files_one_hundred(
|
||||
+ _count_jpg(Path(today_ds, "valid", "images"))
|
||||
+ _count_jpg(Path(today_ds, "test", "images"))
|
||||
)
|
||||
assert n == 100
|
||||
assert n == 20
|
||||
|
||||
|
||||
def test_bt_dsf_04_corrupted_labels_quarantined(
|
||||
@@ -131,7 +131,7 @@ def test_bt_dsf_04_corrupted_labels_quarantined(
|
||||
fixture_labels_dir,
|
||||
):
|
||||
# Arrange
|
||||
stems = [p.stem for p in sorted(fixture_images_dir.glob("*.jpg"))[:100]]
|
||||
stems = [p.stem for p in sorted(fixture_images_dir.glob("*.jpg"))[:20]]
|
||||
corrupt = set(stems[:5])
|
||||
train, today_ds = _prepare_form_dataset(
|
||||
monkeypatch,
|
||||
@@ -139,7 +139,7 @@ def test_bt_dsf_04_corrupted_labels_quarantined(
|
||||
constants_patch,
|
||||
fixture_images_dir,
|
||||
fixture_labels_dir,
|
||||
100,
|
||||
20,
|
||||
corrupt,
|
||||
)
|
||||
# Act
|
||||
@@ -150,7 +150,7 @@ def test_bt_dsf_04_corrupted_labels_quarantined(
|
||||
+ _count_jpg(Path(today_ds, "valid", "images"))
|
||||
+ _count_jpg(Path(today_ds, "test", "images"))
|
||||
)
|
||||
assert split_total == 95
|
||||
assert split_total == 15
|
||||
assert _count_jpg(c_mod.config.corrupted_images_dir) == 5
|
||||
assert len(list(Path(c_mod.config.corrupted_labels_dir).glob("*.txt"))) == 5
|
||||
|
||||
@@ -202,7 +202,7 @@ def test_rl_dsf_02_no_filename_duplication_across_splits(
|
||||
constants_patch,
|
||||
fixture_images_dir,
|
||||
fixture_labels_dir,
|
||||
100,
|
||||
20,
|
||||
set(),
|
||||
)
|
||||
# Act
|
||||
@@ -214,4 +214,4 @@ def test_rl_dsf_02_no_filename_duplication_across_splits(
|
||||
for f in (base / split / "images").glob("*.jpg"):
|
||||
names.append(f.name)
|
||||
assert len(names) == len(set(names))
|
||||
assert len(names) == 100
|
||||
assert len(names) == 20
|
||||
|
||||
Reference in New Issue
Block a user