Advertisement
Guest User

Untitled

a guest
Mar 20th, 2019
57
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.54 KB | None | 0 0
  1. def convert_to_ela_image(path, quality):
  2. filename = path
  3. resaved_filename = filename.split('.')[0] + '.resaved.jpg'
  4. im = Image.open(filename).convert('RGB')
  5. im.save(resaved_filename, 'JPEG', quality=quality)
  6. resaved_im = Image.open(resaved_filename)
  7.  
  8. ela_im = ImageChops.difference(im, resaved_im)
  9.  
  10. extrema = ela_im.getextrema()
  11. max_diff = max([ex[1] for ex in extrema])
  12. if max_diff == 0:
  13. max_diff = 1
  14. scale = 255.0 / max_diff
  15.  
  16. ela_im = ImageEnhance.Brightness(ela_im).enhance(scale)
  17. return ela_im
  18.  
  19. dataset = pd.read_csv('MICC2000.csv')
  20.  
  21. X = []
  22. Y = []
  23.  
  24. for index, row in dataset.iterrows():
  25. X.append(array(convert_to_ela_image(row[0], 90).resize((128, 128))).flatten() / 255.0)
  26. Y.append(row[1])
  27.  
  28. X = np.array(X)
  29. Y = to_categorical(Y, 2)
  30.  
  31. X = X.reshape(-1, 128, 128, 3)
  32.  
  33. X_train, X_val, Y_train, Y_val = train_test_split(X, Y, test_size = 0.50, random_state=5 , shuffle=True)
  34.  
  35.  
  36. model = Sequential()
  37.  
  38.  
  39.  
  40. model.add(Conv2D(filters = 32, kernel_size = (5,5),padding = 'valid', activation ='relu', input_shape = (128,128,3)))
  41. model.add(Conv2D(filters = 64, kernel_size = (5,5), strides=(2,2) ,padding = 'valid', activation ='relu'))
  42. model.add(Conv2D(filters = 128, kernel_size = (5,5),padding = 'valid', activation ='relu'))
  43. model.add(Conv2D(filters = 256, kernel_size = (5,5),strides=(2,2),padding = 'valid', activation ='relu'))
  44. model.add(Dropout(0.25))
  45.  
  46.  
  47. model.add(Flatten())
  48.  
  49.  
  50. model.add(Dense(256, activation = "relu"))
  51. model.add(Dropout(0.5))
  52. model.add(Dense(2, activation = "softmax"))
  53.  
  54. model.summary()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement