From d87b4862a25f9d95079c97412d39bf48b615fdba Mon Sep 17 00:00:00 2001
From: Aymeric Chaumont <aymeric.chaumont@student-cs.fr>
Date: Wed, 20 Jul 2022 16:51:42 +0200
Subject: [PATCH] added doc on the model in README

---
 README.md | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/README.md b/README.md
index c810737..0d8fbf2 100644
--- a/README.md
+++ b/README.md
@@ -44,6 +44,18 @@ Navigate to [http://localhost:3000](http://localhost:3000)
 
 <br/>  
 
+# Documentation
+
+## Algorithm
+
+The crowd-counting AI model used is based on this repository:
+https://github.com/ZhengPeng7/W-Net-Keras
+The dataset used is ShanghaiTech Part B.
+The model is given a 3-channel image and generates a density map of half the size of the input image.
+The estimated number of people is obtained by summing on all pixels of the density map.
+
+<br/>
+
 # TODO
 ## Coté algo
 - Accéder à d'autre infos telle que l'API des cours sur demande à la DISI pour intégrer ça aux prédictions (ex: cours en promo complète juste implique plus d'attente)
-- 
GitLab