|
46 | 46 |
|
47 | 47 | logger.warning("testing with bioimageio.spec {}", bioimageio_spec_version) |
48 | 48 |
|
| 49 | +EXAMPLE_DESCRIPTIONS = "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/" |
| 50 | + |
49 | 51 | # TODO: use models from new collection on S3 |
50 | 52 | MODEL_SOURCES: Dict[str, str] = { |
51 | 53 | "hpa_densenet": "polite-pig/1.1", |
52 | 54 | "stardist": ( |
53 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models" |
54 | | - "/stardist_example_model/v0_4.bioimageio.yaml" |
| 55 | + EXAMPLE_DESCRIPTIONS + "models/stardist_example_model/v0_4.bioimageio.yaml" |
55 | 56 | ), |
56 | 57 | "shape_change": ( |
57 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
58 | | - "upsample_test_model/v0_4.bioimageio.yaml" |
| 58 | + EXAMPLE_DESCRIPTIONS + "models/upsample_test_model/v0_4.bioimageio.yaml" |
59 | 59 | ), |
60 | 60 | "stardist_wrong_shape": ( |
61 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
62 | | - "stardist_example_model/rdf_wrong_shape.yaml" |
| 61 | + EXAMPLE_DESCRIPTIONS + "models/stardist_example_model/rdf_wrong_shape.yaml" |
63 | 62 | ), |
64 | 63 | "stardist_wrong_shape2": ( |
65 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
66 | | - "stardist_example_model/rdf_wrong_shape2_v0_4.yaml" |
| 64 | + EXAMPLE_DESCRIPTIONS |
| 65 | + + "models/stardist_example_model/rdf_wrong_shape2_v0_4.yaml" |
67 | 66 | ), |
68 | 67 | "unet2d_diff_output_shape": ( |
69 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
70 | | - "unet2d_diff_output_shape/v0_4.bioimageio.yaml" |
| 68 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_diff_output_shape/v0_4.bioimageio.yaml" |
71 | 69 | ), |
72 | 70 | "unet2d_expand_output_shape": ( |
73 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
74 | | - "unet2d_nuclei_broad/expand_output_shape.bioimageio.yaml" |
| 71 | + EXAMPLE_DESCRIPTIONS |
| 72 | + + "models/unet2d_nuclei_broad/expand_output_shape.bioimageio.yaml" |
75 | 73 | ), |
76 | 74 | "unet2d_fixed_shape": ( |
77 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
78 | | - "unet2d_fixed_shape/v0_4.bioimageio.yaml" |
| 75 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_fixed_shape/v0_4.bioimageio.yaml" |
79 | 76 | ), |
80 | 77 | "unet2d_keras_tf2": ( |
81 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
82 | | - "unet2d_keras_tf2/v0_4.bioimageio.yaml" |
| 78 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_keras_tf2/v0_4.bioimageio.yaml" |
83 | 79 | ), |
84 | 80 | "unet2d_keras": ( |
85 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
86 | | - "unet2d_keras_tf/v0_4.bioimageio.yaml" |
| 81 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_keras_tf/v0_4.bioimageio.yaml" |
87 | 82 | ), |
88 | 83 | "unet2d_multi_tensor": ( |
89 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
90 | | - "unet2d_multi_tensor/v0_4.bioimageio.yaml" |
| 84 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_multi_tensor/v0_4.bioimageio.yaml" |
91 | 85 | ), |
92 | 86 | "unet2d_nuclei_broad_model_old": ( |
93 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
94 | | - "unet2d_nuclei_broad/v0_4_9.bioimageio.yaml" |
| 87 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_nuclei_broad/v0_4_9.bioimageio.yaml" |
95 | 88 | ), |
96 | 89 | "unet2d_nuclei_broad_model": ( |
97 | | - "https://raw.githubusercontent.com/bioimage-io/spec-bioimage-io/main/example_descriptions/models/" |
98 | | - "unet2d_nuclei_broad/bioimageio.yaml" |
| 90 | + EXAMPLE_DESCRIPTIONS + "models/unet2d_nuclei_broad/bioimageio.yaml" |
99 | 91 | ), |
100 | 92 | } |
101 | 93 |
|
|
0 commit comments