lojzezust / slr Goto Github PK

├── all_list.txt
├── all_weak.yaml
├── all.yaml
├── dextr_masks.json
├── images
├── imus
├── masks_weak
├── objects
├── pa_similarity
├── prior_instance_masks
├── test_list.txt
├── train_list.txt
├── train.yaml
├── val_list.txt
├── val.yaml
└── weak_annotations.json

fl = focal_loss(out['out'], labels['segmentation'], target_scale=self.focal_loss_scale)

separation_loss = torch.tensor(0.0)
if self.separation_loss:
    separation_loss = water_obstacle_separation_loss(
        out['aux'], labels['segmentation'], include_sky=self.separation_loss_sky)

pa_loss = torch.tensor(0.0)
if self.pairwise_affinity_loss:
    ignore_mask = None
    if self.object_loss is not None:
        ignore_mask = labels['objects'].max(1).values
    pa_loss = pairwise_affinity_loss(out['out'], labels['segmentation'], labels['pa_similarity'],
                                     tau=self.pairwise_affinity_loss_tau, target_scale=self.focal_loss_scale, ignore_mask=ignore_mask)

separation_loss = self.separation_loss_lambda * separation_loss
pa_loss = self.pairwise_affinity_loss_lambda *