-
Notifications
You must be signed in to change notification settings - Fork 9
Open
Description
I just installed SpineNet via your guide, and I believe the following packages are missing in your requirements.txt:
- matplotlib
Additionally,
`
class SpineNet:
def __init__(
self, device: bool = "cuda:0", verbose: bool = True, scan_type: str = "lumbar"
) -> None:
"""
Initialize instance of spinenet for (1) detecting and labelling vertebrae
(2) Performing radiological grading for common spinal degenerative changes in T2 sagittal lumbar scans.
Parameters
----------
device : str, optional
The pytorch-style device to use for the model. The default is "cuda:0". If you not using CUDA-enabled machine, you can use "cpu" (although this will slow performance).
`
type mismatch between device being a bool and an optional str.
Best regards
Hendrik
Metadata
Metadata
Assignees
Labels
No labels