src/eval.py [129:141]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                for i in range(pred_k_hots.size(0)):
                    # compute per image metrics
                    image_error_counts = {
                        'tp_c': 0,
                        'fp_c': 0,
                        'fn_c': 0,
                        'tn_c': 0,
                        'tp_all': 0,
                        'fp_all': 0,
                        'fn_all': 0
                    }
                    update_error_counts(image_error_counts, pred_k_hots[i].unsqueeze(0),
                                        target_k_hots[i].unsqueeze(0))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/train.py [350:361]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                        for i in range(pred_k_hots.size(0)):
                            image_error_counts = {
                                'tp_c': 0,
                                'fp_c': 0,
                                'fn_c': 0,
                                'tn_c': 0,
                                'tp_all': 0,
                                'fp_all': 0,
                                'fn_all': 0
                            }
                            update_error_counts(image_error_counts, pred_k_hots[i].unsqueeze(0),
                                                target_k_hots[i].unsqueeze(0))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -