Skip to content

Commit 4a0d139

Browse files
AlexandreBanks6sjh26
authored andcommitted
Update README with progress from project week
Added details about data preprocessing and new data structures.
1 parent f4c874a commit 4a0d139

1 file changed

Lines changed: 74 additions & 0 deletions

File tree

  • PW45_2026_Boston/Projects/AiModelDevelopmentForLungUltrasoundAnalysis

PW45_2026_Boston/Projects/AiModelDevelopmentForLungUltrasoundAnalysis/README.md

Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,8 @@ Therefore, we have started to develop detection models to automatically find ple
7777

7878
1. We have trained some preliminary YOLO models for bounding box detection. We trained one oriented bounding box YOLO model for pleural lines, and another axis-aligned box YOLO model for B-lines.
7979
2. We calculated the percent pleura and compared it to the experts.
80+
3. Implemented and validated a data preprocessing script: 1) parses data and formats into COCO-styled repository, 2) enables sector-to-scanline conversions, 3) enables filtering by metadata (_site, annotator, lung zone, patients, before/after diuretic, transducer type, transducer manufacturer_)
81+
4. Implemented a dataloader that allocates equal proportions of different metadata tags to train/vali/test datasets. Implemented k-fold validation and inclusion of data by various metadata tags.
8082

8183

8284

@@ -95,6 +97,78 @@ YOLO results after performing pleural line detection and B-line detection, and c
9597
Percent pleura for one test lung zone - x = frame #, and y = percentage. We see that the AI predicted
9698
<img width="455" height="333" alt="Image" src="https://github.com/user-attachments/assets/ac226669-a7e1-40dd-b479-cb7cb14a95f0" />
9799

100+
**Data Exploration (During Project Week)**
101+
102+
Overview of different metadata of lung ultrasound dataset:
103+
<img width="1653" height="993" alt="LungUSDataOverview" src="https://github.com/user-attachments/assets/e046e100-d5e3-4aaa-8ee9-0dd1a1b5f8d0" />
104+
105+
Overview of original data structures:
106+
```
107+
+---Andrew/
108+
| +---Andrew-0561119268/
109+
| | 0561119268_08109698.andrew.json
110+
| | 0561119268_08109698.dcm
111+
| | ...
112+
| +---Andrew-1835608883/
113+
| +---Andrew-3594442829/
114+
| +---Andrew-5476058456/
115+
| +---Andrew-5982622832/
116+
| ...
117+
| +---Andrew-CARVD_123_Day_0/
118+
| +---Andrew-CARVD_124_Day_0/
119+
| +---Andrew-CARVD_125_Day_0/
120+
| ...
121+
| +---Andrew-CARVD_134_Day_0/
122+
| +---Andrew-Lahey_001_T0_Transverse/
123+
| +---Andrew-Lahey_029_T0_Transverse/
124+
| +---Andrew-Lahey_032_T0_Transverse/
125+
| ...
126+
+---Arjun/
127+
| ...
128+
+---Frances/
129+
| ...
130+
+---Lao/
131+
| ...
132+
+---Nick/
133+
| ...
134+
+---Nicole/
135+
| ...
136+
+---Peter/
137+
| ...
138+
+---Sandra/
139+
| ...
140+
+---Sandra-old/
141+
...
142+
```
143+
144+
Overview of new data structure:
145+
```
146+
COCO_Data
147+
+---annotations/
148+
| +---scanline/
149+
| | <Annotator>_<clipid>.json
150+
| | <Annotator>_<clipid>.json
151+
| | ...
152+
| +---sector/
153+
| <Annotator>_<clipid>.json
154+
| <Annotator>_<clipid>.json
155+
| ...
156+
+---images/
157+
+---scanline/
158+
| <Annotator>_<clipid>_<framenum>.png
159+
| <Annotator>_<clipid>_<framenum>.png
160+
| ...
161+
+---sector/
162+
<Annotator>_<clipid>_<framenum>.png
163+
<Annotator>_<clipid>_<framenum>.png
164+
...
165+
```
166+
167+
168+
Standardized JSON Metadata:
169+
170+
<img width="600" alt="image" src="https://github.com/user-attachments/assets/70d1a7a3-a83f-4b3e-9629-e7fa67848f1f" />
171+
98172

99173

100174
# Background and References

0 commit comments

Comments
 (0)