yuxin
commited on
Commit
·
ddf6466
1
Parent(s):
d0601bd
add config
Browse files- model_segvol_single.py +1 -1
model_segvol_single.py
CHANGED
|
@@ -520,7 +520,7 @@ class SegVol(nn.Module):
|
|
| 520 |
# organ prompt
|
| 521 |
iter_organs.append(training_organs)
|
| 522 |
# box prompt
|
| 523 |
-
box = generate_box(train_labels[sample_idx])
|
| 524 |
iter_bboxes.append(box)
|
| 525 |
# point prompt
|
| 526 |
num_positive_extra_max, num_negative_extra_max = 10, 10
|
|
|
|
| 520 |
# organ prompt
|
| 521 |
iter_organs.append(training_organs)
|
| 522 |
# box prompt
|
| 523 |
+
box = generate_box(train_labels[sample_idx], bbox_shift=10)
|
| 524 |
iter_bboxes.append(box)
|
| 525 |
# point prompt
|
| 526 |
num_positive_extra_max, num_negative_extra_max = 10, 10
|