Merge pull request #39 from robmarkcole/ISSUE34

fix issue 34

.gitignore

@@ -1,6 +1,5 @@
 venv/
 .vscode
 *.pt
-#*.jpg
-static/tmp.jpg
 *.pyc
+static/*.png

README.md

@@ -6,7 +6,7 @@ Simple app consisting of a form where you can upload an image, and see the infer
 
 `$ python3 webapp.py --port 5000`
 
-then visit http://localhost:5000/ in your browser:
+then visit [http://localhost:5000/](http://localhost:5000/) in your browser:
 
 <p align="center">
 <img src="docs/app_form.jpg" width="450">
@@ -16,6 +16,8 @@ then visit http://localhost:5000/ in your browser:
 <img src="docs/app_result.jpg" width="450">
 </p>
 
+Processed images are saved in the `static` directory with a datetime for the filename.
+
 ## Rest API
 Simple rest API exposing the model for consumption by another service. Run:
 

webapp.py

@@ -6,12 +6,15 @@
 import io
 import os
 from PIL import Image
+import datetime
 
 import torch
 from flask import Flask, render_template, request, redirect
 
 app = Flask(__name__)
 
+DATETIME_FORMAT = "%Y-%m-%d_%H-%M-%S-%f"
+
 
 @app.route("/", methods=["GET", "POST"])
 def predict():
@@ -27,8 +30,10 @@ def predict():
         results = model([img])
 
         results.render()  # updates results.imgs with boxes and labels
-        results.save(save_dir="static/")
-        return redirect("static/image0.jpg")
+        now_time = datetime.datetime.now().strftime(DATETIME_FORMAT)
+        img_savename = f"static/{now_time}.png"
+        Image.fromarray(results.ims[0]).save(img_savename)
+        return redirect(img_savename)
 
     return render_template("index.html")