diff --git a/.idea/misc.xml b/.idea/misc.xml
index 715ed69..8eab719 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -3,5 +3,5 @@
-
+
\ No newline at end of file
diff --git a/.idea/shelf/Uncommitted_changes_before_Update_at_6_8_2020_1_30_PM_[Default_Changelist]/shelved.patch b/.idea/shelf/Uncommitted_changes_before_Update_at_6_8_2020_1_30_PM_[Default_Changelist]/shelved.patch
new file mode 100644
index 0000000..21be4b0
--- /dev/null
+++ b/.idea/shelf/Uncommitted_changes_before_Update_at_6_8_2020_1_30_PM_[Default_Changelist]/shelved.patch
@@ -0,0 +1,24 @@
+Index: .idea/workspace.xml
+IDEA additional info:
+Subsystem: com.intellij.openapi.diff.impl.patch.BaseRevisionTextPatchEP
+<+>\r\n\r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n
\r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n
\r\n \r\n \r\n
\r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n 1589206421672\r\n \r\n \r\n 1589206421672\r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n 1589815443652\r\n \r\n \r\n \r\n 1589815443652\r\n \r\n \r\n 1589816028640\r\n \r\n \r\n \r\n 1589816028640\r\n \r\n \r\n 1589822196451\r\n \r\n \r\n \r\n 1589822196451\r\n \r\n \r\n 1589918562733\r\n \r\n \r\n \r\n 1589918562733\r\n \r\n \r\n 1589952733192\r\n \r\n \r\n \r\n 1589952733192\r\n \r\n \r\n 1589952979012\r\n \r\n \r\n \r\n 1589952979012\r\n \r\n \r\n 1589955873104\r\n \r\n \r\n \r\n 1589955873104\r\n \r\n \r\n 1589964526485\r\n \r\n \r\n \r\n 1589964526485\r\n \r\n \r\n 1589967955430\r\n \r\n \r\n \r\n 1589967955430\r\n \r\n \r\n 1589972251988\r\n \r\n \r\n \r\n 1589972251988\r\n \r\n \r\n 1590359074952\r\n \r\n \r\n \r\n 1590359074952\r\n \r\n \r\n 1590447313737\r\n \r\n \r\n \r\n 1590447313737\r\n \r\n \r\n 1590538529471\r\n \r\n \r\n \r\n 1590538529471\r\n \r\n \r\n 1590846768877\r\n \r\n \r\n \r\n 1590846768877\r\n \r\n \r\n 1590938465634\r\n \r\n \r\n \r\n 1590938465634\r\n \r\n \r\n 1590963692385\r\n \r\n \r\n \r\n 1590963692385\r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n \r\n
+Subsystem: com.intellij.openapi.diff.impl.patch.CharsetEP
+<+>UTF-8
+===================================================================
+--- .idea/workspace.xml (revision 927b8fe54122327a3482d3e6a9b45e268e53df6e)
++++ .idea/workspace.xml (date 1591615797329)
+@@ -257,11 +257,11 @@
+
+
+
+-
+
+
diff --git a/.idea/wozek.iml b/.idea/wozek.iml
index a9742fc..004f809 100644
--- a/.idea/wozek.iml
+++ b/.idea/wozek.iml
@@ -4,10 +4,7 @@
-
+
-
-
-
\ No newline at end of file
diff --git a/coder/model.py b/coder/model.py
index 45772d6..31c0827 100644
--- a/coder/model.py
+++ b/coder/model.py
@@ -1,21 +1,9 @@
-import numpy as np
import torch
-import torchvision
-import matplotlib.pyplot as plt
-from time import time
-from torchvision import datasets, transforms
-from torch import nn, optim
import torch.nn.functional as F
-
-import cv2
-
from nn_model import Net
+from torch import nn, optim
+from torchvision import datasets, transforms
-'''
-Q:
-what is batch?
-
-'''
n_epochs = 3
batch_size_train = 64
batch_size_test = 1000
diff --git a/coder/nn_model.py b/coder/nn_model.py
index d9791f8..acaa1ab 100644
--- a/coder/nn_model.py
+++ b/coder/nn_model.py
@@ -1,6 +1,5 @@
import torch.nn as nn
import torch.nn.functional as F
-import torch.optim as optim
class Net(nn.Module):
diff --git a/coder/rocognizer.py b/coder/rocognizer.py
index 469dd16..3e84c24 100644
--- a/coder/rocognizer.py
+++ b/coder/rocognizer.py
@@ -1,13 +1,7 @@
import cv2
-import matplotlib.pyplot as plt
import torch
-from PIL.Image import Image
-from torch import nn
-from torchvision.transforms import transforms
-from torch.autograd import Variable
-import numpy as np
-
from nn_model import Net
+from torchvision.transforms import transforms
def recognizer(a_path):
@@ -30,9 +24,6 @@ def recognizer(a_path):
rects = [cv2.boundingRect(ctr) for ctr in ctrs]
# load nn model
- input_size = 784 # = 28*28
- hidden_sizes = [128, 128, 64]
- output_size = 10
model = Net()
model.load_state_dict(torch.load('model.pt'))
model.eval()
@@ -60,4 +51,4 @@ def recognizer(a_path):
recognizer("55555.jpg")
-# print(recognizer("55555.jpg"))
+