Update README.md

README.md

@@ -15,29 +15,23 @@ UniformerV2 is a large transformer-based model trained on a binary classificatio
 ### Model Description
 
 <!-- Provide a longer summary of what this model is. -->
-UniformerV2 is a large transformer-based model trained on a binary classification task. Specifically, it is trained to detect whether the input video contains a chimpanzee(s) exhibiting a reaction to the presence of a camera trap. 
+UniformerV2 is a large transformer-based model trained on a binary classification task. Specifically, it is trained to detect whether the input video contains a chimpanzee(s) exhibiting a reaction to the presence of a camera trap. As the dataset heavily favours videos exhibiting no reaction to the camera, we employ a class-balanced focal loss to address the class imbalance.
 
-- **Developed by:** Otto Brookes, Christophe Boesch, 
-- **Model type:** Vision Transformer
+- **Developed by:** Otto Brookes, Christophe Boesch, Hjalmar S. Kühl, Majid Mirmehdi, Tilo Burghardt
+- **Model type:** Vision Transformer, UniformerV2
 - **License:** MIT
-- **Finetuned from model [optional]:** MMAction2, Uniformerv2
 
 ## Training Details
 
 ### Training Data
-
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
 It is trained on camera trap video footage from 15 different countries in Africa as part of the The Pan African Programme: The Cultured Chimpanzee.
 
 ## Evaluation
 
 ### Metrics
-
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
 We use mean average precision to evaluate models 
 
 ### Results
-
 | Dataset   | Model      | Loss       | mAP (%) |
 |-----------|------------|------------|---------|
 | PanAf     | Uniformer  | CB Focal   | 87.82%  |