update readme

Ainimal · Oct 31, 2022 · a553a1c · a553a1c
1 parent 62c3118
commit a553a1c
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -305,6 +305,37 @@ input = torch.ones([3,3,128,128])
 </details>
 
 
+<details>
+<summary> Demo15: Build a 3D DETR model for object detection </summary>
+
+```python
+import wama_modules as ws
+import torch
+
+encoder = ws.resnet(input_channel = 3, per_stage_channel = [8,16,32,64], dim=3)
+decoder = ws.unet(encoder = encoder, output_channel = 3, dim=3)
+
+input = torch.ones([3,3,128,128])
+
+```
+</details>
+
+<details>
+<summary> Demo16: Build a 3D VGG with SE-attention module for multi-instanse classification </summary>
+
+```python
+import wama_modules as ws
+import torch
+
+encoder = ws.resnet(input_channel = 3, per_stage_channel = [8,16,32,64], dim=3)
+decoder = ws.unet(encoder = encoder, output_channel = 3, dim=3)
+
+input = torch.ones([3,3,128,128])
+
+```
+</details>
+
+
 
 ## 5. All modules (or functions)
 

diff --git a/wama_modules/Encoder.py b/wama_modules/Encoder.py
@@ -9,7 +9,7 @@
 
 
 
-
+densenet