mirror of https://github.com/kritiksoman/GIMP-ML
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
10 lines
282 B
Python
10 lines
282 B
Python
4 years ago
|
import cv2
|
||
|
from predictorClass import Predictor
|
||
|
|
||
|
predictor = Predictor(weights_path='best_fpn.h5')
|
||
|
img = cv2.imread('img/img.jpg')
|
||
|
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
||
|
pred = predictor(img, None)
|
||
|
pred = cv2.cvtColor(pred, cv2.COLOR_RGB2BGR)
|
||
|
cv2.imwrite('submit/img.jpg',pred)
|