@@ -444,6 +444,8 @@ <h1>Source code for openvino_xai.methods.black_box.rise</h1><div class="highligh
444
444
< span class ="nb "> super</ span > < span class ="p "> ()</ span > < span class ="o "> .</ span > < span class ="fm "> __init__</ span > < span class ="p "> (</ span >
445
445
< span class ="n "> model</ span > < span class ="o "> =</ span > < span class ="n "> model</ span > < span class ="p "> ,</ span > < span class ="n "> postprocess_fn</ span > < span class ="o "> =</ span > < span class ="n "> postprocess_fn</ span > < span class ="p "> ,</ span > < span class ="n "> preprocess_fn</ span > < span class ="o "> =</ span > < span class ="n "> preprocess_fn</ span > < span class ="p "> ,</ span > < span class ="n "> device_name</ span > < span class ="o "> =</ span > < span class ="n "> device_name</ span >
446
446
< span class ="p "> )</ span >
447
+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> num_masks</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="o "> |</ span > < span class ="kc "> None</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span >
448
+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> num_cells</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="o "> |</ span > < span class ="kc "> None</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span >
447
449
448
450
< span class ="k "> if</ span > < span class ="n "> prepare_model</ span > < span class ="p "> :</ span >
449
451
< span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> prepare_model</ span > < span class ="p "> ()</ span >
@@ -485,13 +487,11 @@ <h1>Source code for openvino_xai.methods.black_box.rise</h1><div class="highligh
485
487
< span class ="sd "> """</ span >
486
488
< span class ="n "> data_preprocessed</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> preprocess_fn</ span > < span class ="p "> (</ span > < span class ="n "> data</ span > < span class ="p "> )</ span >
487
489
488
- < span class ="n "> num_masks</ span > < span class ="p "> ,</ span > < span class ="n "> num_cells</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _preset_parameters</ span > < span class ="p "> (</ span > < span class ="n "> preset</ span > < span class ="p "> ,</ span > < span class ="n "> num_masks</ span > < span class ="p "> ,</ span > < span class ="n "> num_cells</ span > < span class ="p "> )</ span >
490
+ < span class ="bp " > self </ span > < span class =" o " > . </ span > < span class =" n "> num_masks</ span > < span class ="p "> ,</ span > < span class =" bp " > self </ span > < span class =" o " > . </ span > < span class ="n "> num_cells</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _preset_parameters</ span > < span class ="p "> (</ span > < span class ="n "> preset</ span > < span class ="p "> ,</ span > < span class ="n "> num_masks</ span > < span class ="p "> ,</ span > < span class ="n "> num_cells</ span > < span class ="p "> )</ span >
489
491
490
492
< span class ="n "> saliency_maps</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _run_synchronous_explanation</ span > < span class ="p "> (</ span >
491
493
< span class ="n "> data_preprocessed</ span > < span class ="p "> ,</ span >
492
494
< span class ="n "> target_indices</ span > < span class ="p "> ,</ span >
493
- < span class ="n "> num_masks</ span > < span class ="p "> ,</ span >
494
- < span class ="n "> num_cells</ span > < span class ="p "> ,</ span >
495
495
< span class ="n "> prob</ span > < span class ="p "> ,</ span >
496
496
< span class ="n "> seed</ span > < span class ="p "> ,</ span >
497
497
< span class ="p "> )</ span >
@@ -536,8 +536,6 @@ <h1>Source code for openvino_xai.methods.black_box.rise</h1><div class="highligh
536
536
< span class ="bp "> self</ span > < span class ="p "> ,</ span >
537
537
< span class ="n "> data_preprocessed</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span >
538
538
< span class ="n "> target_classes</ span > < span class ="p "> :</ span > < span class ="n "> List</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> |</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
539
- < span class ="n "> num_masks</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
540
- < span class ="n "> num_cells</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
541
539
< span class ="n "> prob</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span >
542
540
< span class ="n "> seed</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
543
541
< span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> :</ span >
@@ -554,8 +552,8 @@ <h1>Source code for openvino_xai.methods.black_box.rise</h1><div class="highligh
554
552
< span class ="n "> rand_generator</ span > < span class ="o "> =</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> random</ span > < span class ="o "> .</ span > < span class ="n "> default_rng</ span > < span class ="p "> (</ span > < span class ="n "> seed</ span > < span class ="o "> =</ span > < span class ="n "> seed</ span > < span class ="p "> )</ span >
555
553
556
554
< span class ="n "> saliency_maps</ span > < span class ="o "> =</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> zeros</ span > < span class ="p "> ((</ span > < span class ="n "> num_targets</ span > < span class ="p "> ,</ span > < span class ="n "> input_size</ span > < span class ="p "> [</ span > < span class ="mi "> 0</ span > < span class ="p "> ],</ span > < span class ="n "> input_size</ span > < span class ="p "> [</ span > < span class ="mi "> 1</ span > < span class ="p "> ]))</ span >
557
- < span class ="k "> for</ span > < span class ="n "> _</ span > < span class ="ow "> in</ span > < span class ="n "> tqdm</ span > < span class ="p "> (</ span > < span class ="nb "> range</ span > < span class ="p "> (</ span > < span class ="mi "> 0</ span > < span class ="p "> ,</ span > < span class ="n "> num_masks</ span > < span class ="p "> ),</ span > < span class ="n "> desc</ span > < span class ="o "> =</ span > < span class ="s2 "> "Explaining in synchronous mode"</ span > < span class ="p "> ):</ span >
558
- < span class ="n "> mask</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _generate_mask</ span > < span class ="p "> (</ span > < span class ="n "> input_size</ span > < span class ="p "> ,</ span > < span class ="n "> num_cells</ span > < span class ="p "> ,</ span > < span class ="n "> prob</ span > < span class ="p "> ,</ span > < span class ="n "> rand_generator</ span > < span class ="p "> )</ span >
555
+ < span class ="k "> for</ span > < span class ="n "> _</ span > < span class ="ow "> in</ span > < span class ="n "> tqdm</ span > < span class ="p "> (</ span > < span class ="nb "> range</ span > < span class ="p "> (</ span > < span class ="mi "> 0</ span > < span class ="p "> ,</ span > < span class ="bp " > self </ span > < span class =" o " > . </ span > < span class =" n "> num_masks</ span > < span class ="p "> ),</ span > < span class ="n "> desc</ span > < span class ="o "> =</ span > < span class ="s2 "> "Explaining in synchronous mode"</ span > < span class ="p "> ):</ span >
556
+ < span class ="n "> mask</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _generate_mask</ span > < span class ="p "> (</ span > < span class ="n "> input_size</ span > < span class ="p "> ,</ span > < span class ="bp " > self </ span > < span class =" o " > . </ span > < span class =" n "> num_cells</ span > < span class ="p "> ,</ span > < span class ="n "> prob</ span > < span class ="p "> ,</ span > < span class ="n "> rand_generator</ span > < span class ="p "> )</ span >
559
557
< span class ="c1 "> # Add channel dimensions for masks</ span >
560
558
< span class ="k "> if</ span > < span class ="n "> is_bhwc_layout</ span > < span class ="p "> (</ span > < span class ="n "> data_preprocessed</ span > < span class ="p "> ):</ span >
561
559
< span class ="n "> masked</ span > < span class ="o "> =</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> expand_dims</ span > < span class ="p "> (</ span > < span class ="n "> mask</ span > < span class ="p "> ,</ span > < span class ="mi "> 2</ span > < span class ="p "> )</ span > < span class ="o "> *</ span > < span class ="n "> data_preprocessed</ span >
0 commit comments