FilesExpand file tree

 master

/

dotdot.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

72 lines (54 loc) · 1.37 KB

 master

/

dotdot.py

Top

File metadata and controls

• Code
• Blame

72 lines (54 loc) · 1.37 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

import cv2

import pandas as pd

import os

import numpy

import io

import cv2

from PIL import Image

import numpy as np

from os import listdir

import skimage.transform

import cv2

import sys

import os

import pickle

from collections import defaultdict

from collections import OrderedDict

import imageio

import skimage

from skimage.io import *

from skimage.transform import *

from collections import OrderedDict

import scipy

import scipy.ndimage as ndimage

import scipy.ndimage.filters as filters

from scipy.ndimage import binary_dilation

import matplotlib.patches as patches

def dot(image_bytes,prediction):

'''

img_folder_path='./00000002_000.png'

frame=cv2.imread(img_folder_path)'''

image = Image.open(io.BytesIO(image_bytes))

frame = np.array(image)

i = prediction[0][4]

j = i.split(' ')[1:]

k = numpy.array(j)

l = k.astype(float)

x1=int(l[0])

y1=int(l[1])

x2=int(l[2])

y2=int(l[3])

a = np.random.randint(9,size=1)

b = np.random.randint(9,size=1)

k = b*10 +a

k = int(k)

k = str(k)

ln = './static/chhati'+k+'.jpg'

cv2.rectangle(frame,(x1,y1),(x2,y2),(0,255,0),3) #rgb 220,20,60

#cv2.rectangle(frame,(x1,y1),(x2,y2),(60,20,220),3)

#print(frame)

cv2.imwrite(ln,frame)

cv2.imshow('image',frame)

cv2.waitKey(10000)

cv2.destroyAllWindows()

return ln