deps: add torch/torchvision/pillow for MRI DL 2D
Browse filesPinned torch==2.4.1, torchvision==0.19.1, pillow==10.4.0 to enable the
2D resnet18 4-class Alzheimer's MRI inference path. These pull no
duplicate transitive deps that conflict with onnxruntime; both backends
coexist for MRI prediction (selected via MRI_MODEL_KIND env in upcoming
commits).
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
- requirements.txt +3 -0
requirements.txt
CHANGED
|
@@ -32,6 +32,9 @@ mlflow==2.16.0
|
|
| 32 |
shap==0.46.0
|
| 33 |
joblib==1.4.2
|
| 34 |
onnxruntime==1.19.2 # MRI volumetric ONNX inference (external DL artifact)
|
|
|
|
|
|
|
|
|
|
| 35 |
|
| 36 |
# --- Tooling / tests ---
|
| 37 |
pytest==8.3.3
|
|
|
|
| 32 |
shap==0.46.0
|
| 33 |
joblib==1.4.2
|
| 34 |
onnxruntime==1.19.2 # MRI volumetric ONNX inference (external DL artifact)
|
| 35 |
+
torch==2.4.1 # MRI 2D resnet18 inference (Plan: external-assets-integration)
|
| 36 |
+
torchvision==0.19.1 # resnet18 + image transforms
|
| 37 |
+
pillow==10.4.0 # image loading for 2D MRI inputs
|
| 38 |
|
| 39 |
# --- Tooling / tests ---
|
| 40 |
pytest==8.3.3
|