Huang Qi
de602365e9
- Move all code into src/mouse_control/ following the src-layout convention, with model/collect/train/visualize/inference as separate modules and SimpleNet extracted into model.py - Add hatchling-based pyproject.toml with three CLI entry points (mouse-collect, mouse-train, mouse-visualize) and bundle the trained ONNX model as a package resource under assets/ - Move training data to data/, delete superseded show.py, remove dead imports (numpy, onnxruntime, onnx, onnxsim) and the unused visualize_path() helper - Fix crashes in the collector: guard against destroyed-widget access after the n==100 destroy(), and skip save_to_csv when recording is off or the path is too short to derive 10 keypoints - Switch ONNX export to external_data=False so the model ships as a single self-contained 19KB file - Bytes-load the bundled model via importlib.resources so packaging remains correct under zip-safe distributions - Rewrite README around the new layout, commands, and public API
20 lines
257 B
Plaintext
20 lines
257 B
Plaintext
# Byte-compiled / build artifacts
|
|
__pycache__/
|
|
*.py[cod]
|
|
*.egg-info/
|
|
build/
|
|
dist/
|
|
|
|
# Virtual envs / tooling
|
|
.venv/
|
|
.uv/
|
|
|
|
# Generated outputs (only at repo root; bundled assets must stay)
|
|
/trajectories.png
|
|
/mouse.onnx
|
|
/mouse.onnx.data
|
|
|
|
# IDE
|
|
.idea/
|
|
.vscode/
|