(p)
| 33 | |
| 34 | |
| 35 | def prompt_switch(p): |
| 36 | p = int(p) |
| 37 | if p == 1: |
| 38 | return 3 |
| 39 | if p == 2: |
| 40 | return 2 |
| 41 | if p == 3: |
| 42 | return 0 |
| 43 | if p == 4: |
| 44 | return 4 |
| 45 | if p == 5: |
| 46 | return 1 |
| 47 | if p == 6: |
| 48 | return 5 |
| 49 | else: |
| 50 | raise NotImplementedError |
| 51 | |
| 52 | |
| 53 | class SemanticSamAutomaticMaskGenerator: |