Moez B commited on
Commit ·
39b09ef
1
Parent(s): f150259
examples
Browse files
app.py
CHANGED
|
@@ -63,8 +63,20 @@ def predict_all_models(img):
|
|
| 63 |
|
| 64 |
#gradio stuff
|
| 65 |
with gr.Blocks() as demo:
|
| 66 |
-
gr.Markdown("#
|
| 67 |
-
gr.Markdown("Upload a single image to see how each segmentation model performs. The first row of output shows the non-generalized basic models. The second row of output shows the better model (U-Net with pretrained ResNet50 encoder) non-generalized and generalized. Generalized = trained on both Laparoscopic Cholecystectomy data and Cataract operation data, Non-generalized = trained only on Cholecystectomy data. This is the free tier so compute power is limited.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
|
| 69 |
with gr.Row():
|
| 70 |
input_image = gr.Image(label="Upload an Image", type="filepath")
|
|
|
|
| 63 |
|
| 64 |
#gradio stuff
|
| 65 |
with gr.Blocks() as demo:
|
| 66 |
+
gr.Markdown("# Generalized Surgical Instrument Segmentation [All Models]")
|
| 67 |
+
gr.Markdown("Upload a single image to see how each segmentation model performs. The first row of output shows the non-generalized basic models. The second row of output shows the better model (U-Net with pretrained ResNet50 encoder) non-generalized and generalized. Generalized = trained on both Laparoscopic Cholecystectomy data and Cataract operation data, Non-generalized = trained only on Cholecystectomy data. This is the free tier so compute power is limited. The only time you may see an error is if multiple people are using it at once. Just refresh and try again if that does happen. By Moez B. and Steven K.")
|
| 68 |
+
|
| 69 |
+
gr.Markdown("Test images to copy and paste in if you don't want to find your own:")
|
| 70 |
+
gr.Markdown("## Laparoscopic Cholecystectomy:")
|
| 71 |
+
gr.Markdown("1. https://medtube.net/images/min/bcdfd6b4e85759ae7dacb7778b679aa6/459/345/1")
|
| 72 |
+
gr.Markdown("2. https://www.sages.org/wp-content/uploads/2019/09/fecXdNs6rp0.jpg")
|
| 73 |
+
gr.Markdown("3. https://medtube.net/images/min/6041ca08cd4d11099559472fa4cd992f/459/345/1")
|
| 74 |
+
|
| 75 |
+
gr.Markdown("## Cataract Surgery:")
|
| 76 |
+
gr.Markdown("1. https://core4-cms.imgix.net/Screen%20Shot%202021-12-13%20at%203.41.12%20PM_1639428086.png")
|
| 77 |
+
gr.Markdown("2. https://core4-cms.imgix.net/Screen%20Shot%202021-11-30%20at%2010.31.36%20AM_1638286314.png")
|
| 78 |
+
gr.Markdown("3. https://i.ytimg.com/vi/iFQ9-i41Z0I/sddefault.jpg")
|
| 79 |
+
|
| 80 |
|
| 81 |
with gr.Row():
|
| 82 |
input_image = gr.Image(label="Upload an Image", type="filepath")
|