Spaces:
Sleeping
Sleeping
crazyscientist
commited on
Commit
•
6ce5c48
1
Parent(s):
b767afd
Update modInference.py
Browse files- modInference.py +1 -3
modInference.py
CHANGED
|
@@ -3,8 +3,6 @@ import cv2
|
|
| 3 |
import torch
|
| 4 |
import glob as glob
|
| 5 |
import os
|
| 6 |
-
import time
|
| 7 |
-
|
| 8 |
import matplotlib.pyplot as plt
|
| 9 |
|
| 10 |
from models.create_fasterrcnn_model import create_model
|
|
@@ -47,7 +45,7 @@ top_model = Dropout(0.2)(top_model)
|
|
| 47 |
output_layer = Dense(5, activation='softmax')(top_model)
|
| 48 |
|
| 49 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|
| 50 |
-
CNN.load_weights("
|
| 51 |
|
| 52 |
|
| 53 |
|
|
|
|
| 3 |
import torch
|
| 4 |
import glob as glob
|
| 5 |
import os
|
|
|
|
|
|
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
|
| 8 |
from models.create_fasterrcnn_model import create_model
|
|
|
|
| 45 |
output_layer = Dense(5, activation='softmax')(top_model)
|
| 46 |
|
| 47 |
CNN = Model(inputs=conv_base.input, outputs=output_layer)
|
| 48 |
+
CNN.load_weights("CNN.hdf5")
|
| 49 |
|
| 50 |
|
| 51 |
|