Dave-A commited on
Commit
fd651b5
·
1 Parent(s): 0d7fd9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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 cv2
 
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)