ToolArena/tasks/retfound_feature_vector/task.yaml at main · KatherLab/ToolArena · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
name: retfound_feature_vector
repo:
  name: RETFound
  url: "https://github.com/rmaphoh/RETFound_MAE"
  commit: 897d71c
  env:
    - name: HF_TOKEN
      value: "${env:HF_TOKEN}"
papers: [zhou2023retfound]
category: misc  # imaging
requires: cuda
description: >-
  Extract the latent feature vector for the given retinal image using the RETFound pretrained RETFound_mae_natureCFP model.
arguments:
  - name: image_file
    description: Path to the retinal image.
    type: str
returns:
  - name: feature_vector
    description: The feature vector for the given retinal image, as a list of floats.
    type: list
example:
  arguments:
    - name: image_file
      value: /mount/input/retinal_image.jpg
  mount:
    - source: cucumber.jpg
      target: retinal_image.jpg
test_invocations:
  - name: jpg
    arguments:
      - name: image_file
        value: /mount/input/image1.jpg
    mount:
      - source: "TCGA-BRCA_patch_TCGA-BH-A0DE-01Z-00-DX1.64A0340A-8146-48E8-AAF7-4035988B9152.jpg"
        target: image1.jpg
  - name: png
    arguments:
      - name: image_file
        value: /mount/input/image2.png
    mount:
      - source: "TCGA-BRCA_patch_TCGA-BH-A0DE-01Z-00-DX1.64A0340A-8146-48E8-AAF7-4035988B9152.png"
        target: image2.png
  - name: cucumber_different_filename
    arguments:
      - name: image_file
        value: /mount/input/cucumber.jpg
    mount:
      - source: "cucumber.jpg"
        target: cucumber.jpg