Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,7 @@ license: mit
|
|
| 12 |
|
| 13 |
# ERA-SESSION13 YoloV3 with Pytorch Lightning & Gradio
|
| 14 |
|
| 15 |
-
HF Link: https://huggingface.co/spaces/
|
| 16 |
|
| 17 |
### Achieved:
|
| 18 |
1. **Training Loss: 3.680**
|
|
@@ -21,13 +21,42 @@ HF Link: https://huggingface.co/spaces/RaviNaik/ERA-SESSION13
|
|
| 21 |
4. **No obj accuracy: 97.991463%**
|
| 22 |
5. **Obj accuracy: 75.976616%**
|
| 23 |
6. **MAP: 0.4366795**
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
### Results
|
| 26 |
-

|
| 501 |
-

|
| 31 |
+
- Ideally trained till 40 epochs
|
| 32 |
+
3. :heavy_check_mark: Add these training features:
|
| 33 |
+
- Add multi-resolution training - the code shared trains only on one resolution 416
|
| 34 |
+
- Add Implement Mosaic Augmentation only 75% of the times
|
| 35 |
+
- Train on float16
|
| 36 |
+
- GradCam must be implemented.
|
| 37 |
+
4. :heavy_check_mark: Things that are allowed due to HW constraints:
|
| 38 |
+
- Change of batch size
|
| 39 |
+
- Change of resolution
|
| 40 |
+
- Change of OCP parameters
|
| 41 |
+
5. :heavy_check_mark: Once done:
|
| 42 |
+
- Move the app to HuggingFace Spaces
|
| 43 |
+
- Allow custom upload of images
|
| 44 |
+
- Share some samples from the existing dataset
|
| 45 |
+
- Show the GradCAM output for the image that the user uploads as well as for the samples.
|
| 46 |
+
6. :heavy_check_mark: Mention things like:
|
| 47 |
+
- classes that your model support
|
| 48 |
+
- link to the actual model
|
| 49 |
+
7. :heavy_check_mark: Assignment:
|
| 50 |
+
- Share HuggingFace App Link
|
| 51 |
+
- Share LightningCode Link on Github
|
| 52 |
+
- Share notebook link (with logs) on GitHub
|
| 53 |
|
| 54 |
### Results
|
| 55 |
+

|
| 56 |
|
| 57 |
### Gradio App
|
| 58 |
+

|
| 59 |
+

|
| 60 |
|
| 61 |
### Model Summary
|
| 62 |
```python
|
|
|
|
| 426 |
```
|
| 427 |
|
| 428 |
### LR Finder
|
| 429 |
+

|
| 430 |
|
| 431 |
### Loss & Accuracy
|
| 432 |
**Training & Validation Loss:**
|
| 433 |
+

|
| 434 |
|
| 435 |
**Testing Accuracy:**
|
| 436 |
```python
|
|
|
|
| 523 |
MAP: 0.43667954206466675
|
| 524 |
```
|
| 525 |
### Tensorboard Plots
|
| 526 |
+
**Training Loss vs Steps:** 
|
| 527 |
|
| 528 |
**Validation Loss vs Steps:**
|
| 529 |
(Info: Validation loss calculated every 10 epochs to save time, thats why the straight line)
|
| 530 |
+

|
| 531 |
|
| 532 |
### GradCAM Representations
|
| 533 |
EigenCAM is used to generate CAM representation, since usal gradient based method wont work with detection models like Yolo, FRCNN etc.
|
| 534 |
+

|
| 535 |
+
|
| 536 |
+
|