hwanglab / DGP_RF_WSL_TripletLoss

DGP_RF_WSL_TripletLoss

1. model_dgp_rf_WSL_triplet.py (The initial DGP model with the triplet loss)
2. model_dgp_rf_WSL_triplet_prob.py (The DGP model with the probabilistic triplet loss [1])

[1] Frederik Warburg, Martin Jørgensen, Javier Civera, Søren Hauberg, Bayesian Triplet Loss: Uncertainty Quantification in Image Retrieval, 2021 ICCV