Update .gitignore

.gitignore

@@ -1,2 +1,5 @@
 
 .DS_Store
+.env
+.pt
+yolov7-e6e.pt

2.4.1'

@@ -0,0 +1 @@
+Defaulting to user installation because normal site-packages is not writeable

4.1.1'

@@ -0,0 +1 @@
+Defaulting to user installation because normal site-packages is not writeable

app.py

@@ -1,12 +1,14 @@
+import base64
 import io
 from operator import truediv
 import os
 import json
 from PIL import Image
-
+import logging
+import numpy as np
 import torch
-from flask import Flask, jsonify, url_for, render_template, request, redirect
-
+from flask import Flask, jsonify, url_for, render_template, request, redirect,send_file
+from PIL import Image as im
 app = Flask(__name__)
 
 RESULT_FOLDER = os.path.join('static')
@@ -20,7 +22,7 @@ def find_model():
     print("please place a model file in this directory!")
 
 model_name = find_model()
-model =torch.hub.load("WongKinYiu/yolov7", 'custom',model_name)
+model =torch.hub.load("ultralytics/yolov5", 'custom',model_name)
 
 model.eval()
 
@@ -48,6 +50,29 @@ def predict():
         return render_template('result.html',result_image = filename,model_name = model_name)
 
     return render_template('index.html')
+@app.route('/check', methods=['GET', 'POST'])
+def predict_check():
+     if request.method == 'POST':
+        if 'file' not in request.files:
+            return redirect(request.url)
+        file = request.files.get('file')
+        if not file:
+            return
+
+        img_bytes = file.read()
+        results = get_prediction(img_bytes)
+        imagecheck = np.squeeze(results.render())
+        #encoded_string= base64.b64encode(imagecheck)
+        data = im.fromarray(imagecheck)
+        encoded_string= base64.b64encode(data.tobytes())
+        #app.logger.info(print(type(encoded_string)))
+        response = {
+            'Status': 'Success',
+            'message': 'Success',
+            'ImageBytes': str(encoded_string)
+        }
+        return jsonify(response)
+
 @app.route('/detect', methods=['GET', 'POST'])
 def handle_video():
     # some code to be implemented later

data (1).yaml

@@ -0,0 +1,24 @@
+names:
+- African_hunting_dog
+- Chihuahua
+- Lhasa
+- Pembroke
+- Pomeranian
+- Saint_Bernard
+- Yorkshire_terrier
+- basset
+- beagle
+- borzoi
+- boxer
+- chow
+- collie
+- dingo
+- golden_retriever
+- malamute
+- malinois
+- papillon
+- pug
+- standard_schnauzer
+nc: 20
+train: Doggos-1/train/images
+val: Doggos-1/valid/images

data.yaml

@@ -0,0 +1,24 @@
+names:
+- African_hunting_dog
+- Chihuahua
+- Lhasa
+- Pembroke
+- Pomeranian
+- Saint_Bernard
+- Yorkshire_terrier
+- basset
+- beagle
+- borzoi
+- boxer
+- chow
+- collie
+- dingo
+- golden_retriever
+- malamute
+- malinois
+- papillon
+- pug
+- standard_schnauzer
+nc: 20
+train: Doggos-1/train/images
+val: Doggos-1/valid/images

opt.yaml

@@ -0,0 +1,37 @@
+weights: ''
+cfg: ./models/custom_yolov5s.yaml
+data: /kaggle/working/yolov5/Doggos-1/data.yaml
+hyp: data/hyp.scratch.yaml
+epochs: 350
+batch_size: 32
+img_size:
+- 416
+- 416
+rect: false
+resume: false
+nosave: false
+notest: false
+noautoanchor: false
+evolve: false
+bucket: ''
+cache_images: true
+image_weights: false
+device: ''
+multi_scale: false
+single_cls: false
+adam: false
+sync_bn: false
+local_rank: -1
+log_imgs: 16
+log_artifacts: false
+workers: 8
+project: runs/train
+entity: null
+name: yolov5s_results
+exist_ok: false
+quad: false
+linear_lr: false
+world_size: 1
+global_rank: -1
+save_dir: runs/train/yolov5s_results
+total_batch_size: 32