|
761 | 761 | "test_images = sorted(glob.glob(os.path.join(data_dir, \"imagesTs\", \"*.nii.gz\")))\n", |
762 | 762 | "test_data = [{\"image\": image, \"label_prompt\": [3]} for image in test_images]\n", |
763 | 763 | "\n", |
764 | | - "from vista3d_config import VISTA3DConfig\n", |
765 | | - "from vista3d_model import VISTA3DModel\n", |
766 | | - "from vista3d_pipeline import VISTA3DPipeline\n", |
| 764 | + "# These three are from the vista3d hugging face model. Can only be imported after downloading.\n", |
| 765 | + "from vista3d_config import VISTA3DConfig # noqa: E402\n", |
| 766 | + "from vista3d_model import VISTA3DModel # noqa: E402\n", |
| 767 | + "from vista3d_pipeline import VISTA3DPipeline # noqa: E402\n", |
767 | 768 | "\n", |
768 | 769 | "config = VISTA3DConfig()\n", |
769 | 770 | "vista3d_hf_model = VISTA3DModel(config)\n", |
|
777 | 778 | }, |
778 | 779 | { |
779 | 780 | "cell_type": "code", |
780 | | - "execution_count": 20, |
| 781 | + "execution_count": null, |
781 | 782 | "metadata": {}, |
782 | 783 | "outputs": [ |
783 | 784 | { |
|
806 | 807 | "numpy_transform = ToNumpy()\n", |
807 | 808 | "loader = LoadImage()\n", |
808 | 809 | "\n", |
| 810 | + "\n", |
809 | 811 | "def convert_to_numpy(input):\n", |
810 | 812 | " numpy_output = numpy_transform(input)\n", |
811 | 813 | " numpy_output = np.squeeze(numpy_output, axis=0)\n", |
|
0 commit comments