0

src/image_recognition/LICENSE

@@ -10,7 +10,7 @@ GNU GENERAL PUBLIC LICENSE
   The GNU General Public License is a free, copyleft license for
 software and other kinds of works.
 
-  The licenses for most software and other practical works are designed
+  The licenses for most software and other practical works are designed 
 to take away your freedom to share and change the works.  By contrast,
 the GNU General Public License is intended to guarantee your freedom to
 share and change all versions of a program--to make sure it remains free

src/image_recognition/data/data.yaml

@@ -10,3 +10,4 @@ nc: 2
 
 # class names
 names: ['phone', 'person']
+ 

src/image_recognition/data_gen.py

@@ -19,7 +19,7 @@ import os
 import shutil
 from pathlib import Path
 from shutil import copyfile
-
+ 
 import cv2
 from PIL import Image, ImageDraw
 from xml.dom.minidom import parse

src/image_recognition/detect.py

@@ -14,7 +14,7 @@ Usage:
 
 import argparse
 import os
-import sys
+import sys 
 from pathlib import Path
 
 import cv2

src/image_recognition/export.py

@@ -13,7 +13,7 @@ Inference:
                                          yolov5s.pb
                                          yolov5s.tflite
 
-TensorFlow.js:
+TensorFlow.js: 
     $ cd .. && git clone https://github.com/zldrobit/tfjs-yolov5-example.git && cd tfjs-yolov5-example
     $ npm install
     $ ln -s ../../yolov5/yolov5s_web_model public/yolov5s_web_model

src/image_recognition/hubconf.py

@@ -5,7 +5,7 @@ PyTorch Hub models https://pytorch.org/hub/ultralytics_yolov5/
 Usage:
     import torch
     model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
-"""
+""" 
 
 import torch
 

src/image_recognition/requirements.txt

@@ -33,4 +33,4 @@ seaborn>=0.11.0
 # Cython  # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
 # pycocotools>=2.0  # COCO mAP
 # roboflow
-thop  # FLOPs computation
+thop  # FLOPs computation 

src/image_recognition/setup.cfg

@@ -17,6 +17,7 @@ addopts =
     --color=yes
 
 
+
 [flake8]
 max-line-length = 120
 exclude = .tox,*.egg,build,temp

src/image_recognition/utils/activations.py

@@ -15,7 +15,7 @@ class SiLU(nn.Module):  # export-friendly version of nn.SiLU()
         return x * torch.sigmoid(x)
 
 
-class Hardswish(nn.Module):  # export-friendly version of nn.Hardswish()
+class Hardswish(nn.Module):  # export-friendly version of nn.Hardswish() 
     @staticmethod
     def forward(x):
         # return x * F.hardsigmoid(x)  # for torchscript and CoreML