Skip to content

Commit

Permalink
image size fix #2 in readme
Browse files Browse the repository at this point in the history
  • Loading branch information
KatHellg committed Sep 24, 2024
1 parent 1ebadd9 commit 6c3bb75
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions examples/welding-defect-detection/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,11 @@

This is an example FEDn project that trains a YOLOv8n model on images of welds to classify them as "good", "bad", or "defected". The dataset is pre-labeled and can be accessed for free from Kaggle https://www.kaggle.com/datasets/sukmaadhiwijaya/welding-defect-object-detection. See a few examples below,

<img src="figs/fig1.jpg" width=50% height=50%>
<img src="figs/fig1.jpg" width=30% height=30%>

<img src="figs/fig2.jpg" width=50% height=50%>
<img src="figs/fig2.jpg" width=30% height=30%>

<img src="figs/fig3.jpg" width=50% height=50%>
<img src="figs/fig3.jpg" width=30% height=30%>


This example is generalizable to many manufacturing and operations use cases, such as automatic optical inspection. The federated setup enables the organization to make use of available data in different factories and in different parts of the manufacturing process, without having to centralize the data.
Expand Down Expand Up @@ -106,14 +106,14 @@ Approach: The number of epochs and rounds in each experiment are divided such th

Centralized:

<img src="figs/CentralizedmAP50.png" width=70% height=70%>
<img src="figs/CentralizedmAP50.png" width=50% height=50%>


Federated:

<img src="figs/2clients_5epochs_50rounds.png" width=30% height=30%>
<img src="figs/2clients_5epochs_50rounds.png" width=50% height=50%>

<img src="figs/2clients_10epochs_25rounds.png" width=30% height=30%>
<img src="figs/2clients_10epochs_25rounds.png" width=50% height=50%>

<img src="figs/2clients_25epochs_10rounds.png" width=30% height=30%>
<img src="figs/2clients_25epochs_10rounds.png" width=50% height=50%>

0 comments on commit 6c3bb75

Please sign in to comment.