Commit
·
cf41f6a
1
Parent(s):
042c19a
Delete Examples
Browse files- Examples/000001.jpg +0 -0
- Examples/000002.jpg +0 -0
- Examples/000003.jpg +0 -0
- Examples/000004.jpg +0 -0
- Examples/000005.jpg +0 -0
- Examples/000006.jpg +0 -0
- Examples/000007.jpg +0 -0
- Examples/000008.jpg +0 -0
- Examples/000009.jpg +0 -0
- Examples/000010.jpg +0 -0
- Examples/000011.jpg +0 -0
- Examples/000012.jpg +0 -0
- Examples/000013.jpg +0 -0
- Examples/000014.jpg +0 -0
- Examples/000015.jpg +0 -0
- Examples/000016.jpg +0 -0
- Examples/000017.jpg +0 -0
- Examples/000018.jpg +0 -0
- Examples/000019.jpg +0 -0
- Examples/t.txt +0 -0
- Examples/train.py +0 -23
Examples/000001.jpg
DELETED
|
Binary file (78.8 kB)
|
|
|
Examples/000002.jpg
DELETED
|
Binary file (113 kB)
|
|
|
Examples/000003.jpg
DELETED
|
Binary file (123 kB)
|
|
|
Examples/000004.jpg
DELETED
|
Binary file (103 kB)
|
|
|
Examples/000005.jpg
DELETED
|
Binary file (85 kB)
|
|
|
Examples/000006.jpg
DELETED
|
Binary file (79.8 kB)
|
|
|
Examples/000007.jpg
DELETED
|
Binary file (115 kB)
|
|
|
Examples/000008.jpg
DELETED
|
Binary file (82.7 kB)
|
|
|
Examples/000009.jpg
DELETED
|
Binary file (137 kB)
|
|
|
Examples/000010.jpg
DELETED
|
Binary file (106 kB)
|
|
|
Examples/000011.jpg
DELETED
|
Binary file (88.1 kB)
|
|
|
Examples/000012.jpg
DELETED
|
Binary file (56.9 kB)
|
|
|
Examples/000013.jpg
DELETED
|
Binary file (134 kB)
|
|
|
Examples/000014.jpg
DELETED
|
Binary file (101 kB)
|
|
|
Examples/000015.jpg
DELETED
|
Binary file (90.2 kB)
|
|
|
Examples/000016.jpg
DELETED
|
Binary file (121 kB)
|
|
|
Examples/000017.jpg
DELETED
|
Binary file (95.6 kB)
|
|
|
Examples/000018.jpg
DELETED
|
Binary file (52.8 kB)
|
|
|
Examples/000019.jpg
DELETED
|
Binary file (58.5 kB)
|
|
|
Examples/t.txt
DELETED
|
File without changes
|
Examples/train.py
DELETED
|
@@ -1,23 +0,0 @@
|
|
| 1 |
-
"""
|
| 2 |
-
Main file for training Yolo model on Pascal VOC and COCO dataset
|
| 3 |
-
"""
|
| 4 |
-
|
| 5 |
-
import torch
|
| 6 |
-
from pytorch_lightning import LightningModule, Trainer, seed_everything
|
| 7 |
-
import torch
|
| 8 |
-
|
| 9 |
-
import warnings
|
| 10 |
-
warnings.filterwarnings("ignore")
|
| 11 |
-
|
| 12 |
-
torch.backends.cudnn.benchmark = True
|
| 13 |
-
|
| 14 |
-
class YOLOTraining(LightningModule):
|
| 15 |
-
def __init__(self,model):
|
| 16 |
-
super().__init__()
|
| 17 |
-
self.model = model
|
| 18 |
-
|
| 19 |
-
def forward(self, x):
|
| 20 |
-
return self.model(x)
|
| 21 |
-
|
| 22 |
-
if __name__ == "__main__":
|
| 23 |
-
num_classes = 20
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|