(self,
train_dataset,
test_dataset,
model,
loss,
y_target,
poisoned_rate,
reflection_candidates,
poisoned_transform_train_index=0,
poisoned_transform_test_index=0,
poisoned_target_transform_index=0,
schedule=None,
seed=0,
deterministic=False,
max_image_size=560,
ghost_rate=0.49,
alpha_b=-1.,
offset=(0, 0),
sigma=-1,
ghost_alpha=-1.)
| 504 | """ |
| 505 | |
| 506 | def __init__(self, |
| 507 | train_dataset, |
| 508 | test_dataset, |
| 509 | model, |
| 510 | loss, |
| 511 | y_target, |
| 512 | poisoned_rate, |
| 513 | reflection_candidates, |
| 514 | poisoned_transform_train_index=0, |
| 515 | poisoned_transform_test_index=0, |
| 516 | poisoned_target_transform_index=0, |
| 517 | schedule=None, |
| 518 | seed=0, |
| 519 | deterministic=False, |
| 520 | max_image_size=560, |
| 521 | ghost_rate=0.49, |
| 522 | alpha_b=-1., |
| 523 | offset=(0, 0), |
| 524 | sigma=-1, |
| 525 | ghost_alpha=-1.): |
| 526 | |
| 527 | super(Refool, self).__init__( |
| 528 | train_dataset=train_dataset, |
| 529 | test_dataset=test_dataset, |
| 530 | model=model, |
| 531 | loss=loss, |
| 532 | schedule=schedule, |
| 533 | seed=seed, |
| 534 | deterministic=deterministic) |
| 535 | |
| 536 | self.poisoned_train_dataset = CreatePoisonedDataset( |
| 537 | train_dataset, |
| 538 | y_target, |
| 539 | poisoned_rate, |
| 540 | poisoned_transform_train_index, |
| 541 | poisoned_target_transform_index, |
| 542 | reflection_candidates, |
| 543 | max_image_size, |
| 544 | ghost_rate, |
| 545 | alpha_b, |
| 546 | offset, |
| 547 | sigma, |
| 548 | ghost_alpha) |
| 549 | |
| 550 | self.poisoned_test_dataset = CreatePoisonedDataset( |
| 551 | test_dataset, |
| 552 | y_target, |
| 553 | 1.0, |
| 554 | poisoned_transform_test_index, |
| 555 | poisoned_target_transform_index, |
| 556 | reflection_candidates, |
| 557 | max_image_size, |
| 558 | ghost_rate, |
| 559 | alpha_b, |
| 560 | offset, |
| 561 | sigma, |
| 562 | ghost_alpha) |
no test coverage detected