RashiAgarwal
commited on
Commit
·
f397a01
1
Parent(s):
2a4462d
Update config.py
Browse files
config.py
CHANGED
@@ -3,7 +3,7 @@ import cv2
|
|
3 |
import torch
|
4 |
|
5 |
from albumentations.pytorch import ToTensorV2
|
6 |
-
from utils import seed_everything
|
7 |
|
8 |
DATASET = '/content/PASCAL_VOC'
|
9 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
@@ -103,85 +103,3 @@ PASCAL_CLASSES = [
|
|
103 |
"train",
|
104 |
"tvmonitor"
|
105 |
]
|
106 |
-
|
107 |
-
COCO_LABELS = ['person',
|
108 |
-
'bicycle',
|
109 |
-
'car',
|
110 |
-
'motorcycle',
|
111 |
-
'airplane',
|
112 |
-
'bus',
|
113 |
-
'train',
|
114 |
-
'truck',
|
115 |
-
'boat',
|
116 |
-
'traffic light',
|
117 |
-
'fire hydrant',
|
118 |
-
'stop sign',
|
119 |
-
'parking meter',
|
120 |
-
'bench',
|
121 |
-
'bird',
|
122 |
-
'cat',
|
123 |
-
'dog',
|
124 |
-
'horse',
|
125 |
-
'sheep',
|
126 |
-
'cow',
|
127 |
-
'elephant',
|
128 |
-
'bear',
|
129 |
-
'zebra',
|
130 |
-
'giraffe',
|
131 |
-
'backpack',
|
132 |
-
'umbrella',
|
133 |
-
'handbag',
|
134 |
-
'tie',
|
135 |
-
'suitcase',
|
136 |
-
'frisbee',
|
137 |
-
'skis',
|
138 |
-
'snowboard',
|
139 |
-
'sports ball',
|
140 |
-
'kite',
|
141 |
-
'baseball bat',
|
142 |
-
'baseball glove',
|
143 |
-
'skateboard',
|
144 |
-
'surfboard',
|
145 |
-
'tennis racket',
|
146 |
-
'bottle',
|
147 |
-
'wine glass',
|
148 |
-
'cup',
|
149 |
-
'fork',
|
150 |
-
'knife',
|
151 |
-
'spoon',
|
152 |
-
'bowl',
|
153 |
-
'banana',
|
154 |
-
'apple',
|
155 |
-
'sandwich',
|
156 |
-
'orange',
|
157 |
-
'broccoli',
|
158 |
-
'carrot',
|
159 |
-
'hot dog',
|
160 |
-
'pizza',
|
161 |
-
'donut',
|
162 |
-
'cake',
|
163 |
-
'chair',
|
164 |
-
'couch',
|
165 |
-
'potted plant',
|
166 |
-
'bed',
|
167 |
-
'dining table',
|
168 |
-
'toilet',
|
169 |
-
'tv',
|
170 |
-
'laptop',
|
171 |
-
'mouse',
|
172 |
-
'remote',
|
173 |
-
'keyboard',
|
174 |
-
'cell phone',
|
175 |
-
'microwave',
|
176 |
-
'oven',
|
177 |
-
'toaster',
|
178 |
-
'sink',
|
179 |
-
'refrigerator',
|
180 |
-
'book',
|
181 |
-
'clock',
|
182 |
-
'vase',
|
183 |
-
'scissors',
|
184 |
-
'teddy bear',
|
185 |
-
'hair drier',
|
186 |
-
'toothbrush'
|
187 |
-
]
|
|
|
3 |
import torch
|
4 |
|
5 |
from albumentations.pytorch import ToTensorV2
|
6 |
+
#from utils import seed_everything
|
7 |
|
8 |
DATASET = '/content/PASCAL_VOC'
|
9 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
103 |
"train",
|
104 |
"tvmonitor"
|
105 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|