khalednabawi11 commited on
Commit
b81fe05
·
verified ·
1 Parent(s): a572204

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -0
README.md ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - hongrui/mimic_chest_xray_v_1
4
+ ---
5
+ # 🩺 CheXNet-MedScan-Report-Gen
6
+
7
+ **CheXNet-MedScan-Report-Gen** is an image captioning model for generating diagnostic text reports from chest X-ray images. It combines the power of a pretrained CheXNet encoder (based on DenseNet121) and a bidirectional LSTM decoder to produce sequence-based textual descriptions.
8
+
9
+ ---
10
+
11
+ ## 🧠 Model Architecture
12
+
13
+ - **Encoder:** DenseNet121 (CheXNet) with classifier removed
14
+ - **Decoder:** Bidirectional LSTM with dropout
15
+ - **Feature dimension:** 1024
16
+ - **Embedding dimension:** 256
17
+ - **Hidden dimension:** 512
18
+ - **Vocabulary size:** 5000
19
+ - **Dropout:** 0.5
20
+
21
+ ---
22
+
23
+ ## 🔧 Usage
24
+
25
+ You can load the model using the Hugging Face Transformers library:
26
+
27
+ ```python
28
+ from transformers import AutoTokenizer, AutoConfig
29
+ from modeling import ImageCaptioningModel
30
+
31
+ config = AutoConfig.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen")
32
+ model = ImageCaptioningModel.from_pretrained("khalednabawi11/Chexnet-MedScan-Report-Gen", config=config)