RashiAgarwal commited on
Commit
f397a01
·
1 Parent(s): 2a4462d

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -83
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
  ]