Moez B commited on
Commit
39b09ef
·
1 Parent(s): f150259
Files changed (1) hide show
  1. app.py +14 -2
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("## 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. If lots of people use at once it may give an error. Just wait a bit and try again. By Moez B. and Steven K.")
 
 
 
 
 
 
 
 
 
 
 
 
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")