Update app.py
Browse files
app.py
CHANGED
@@ -15,11 +15,12 @@ from torch.utils.data import Dataset, DataLoader
|
|
15 |
#from sklearn.model_selection import StratifiedKFold
|
16 |
#from sklearn.metrics import accuracy_score
|
17 |
|
18 |
-
from tqdm import tqdm
|
19 |
-
import
|
|
|
20 |
#import albumentations as A # Albumentations is a computer vision tool that boosts the performance of deep convolutional neural networks. (https://albumentations.ai/)
|
21 |
-
import matplotlib.pyplot as plt
|
22 |
-
import seaborn as sns
|
23 |
#from albumentations.pytorch.transforms import ToTensorV2
|
24 |
|
25 |
model = models.resnet50(pretrained=False)
|
|
|
15 |
#from sklearn.model_selection import StratifiedKFold
|
16 |
#from sklearn.metrics import accuracy_score
|
17 |
|
18 |
+
#from tqdm import tqdm
|
19 |
+
import opencv-python
|
20 |
+
#import cv2
|
21 |
#import albumentations as A # Albumentations is a computer vision tool that boosts the performance of deep convolutional neural networks. (https://albumentations.ai/)
|
22 |
+
#import matplotlib.pyplot as plt
|
23 |
+
#import seaborn as sns
|
24 |
#from albumentations.pytorch.transforms import ToTensorV2
|
25 |
|
26 |
model = models.resnet50(pretrained=False)
|