Update README.md
Browse files
README.md
CHANGED
@@ -12,12 +12,13 @@ license: apache-2.0
|
|
12 |
|
13 |
|
14 |
|
|
|
15 |
|
16 |
## Project File Structure
|
17 |
|
18 |
```
|
19 |
KB-VQA
|
20 |
-
βββ Files: Various files required for the demo such as samples images
|
21 |
βββ models
|
22 |
| βββ deformable-detr-detic: DETIC Object Detection Model.
|
23 |
| βββ yolov5: YOLOv5 Object Detection Model.baseline)
|
@@ -34,20 +35,20 @@ KB-VQA
|
|
34 |
β β βββ fine_tuner.py: Main Fine-Tuning Script for LLaMa-2 Chat models.
|
35 |
β β βββ fine_tuning_data_handler.py: Handles and prepares the data for fine-tuning LLaMA-2 Chat models.
|
36 |
β β βββ fine_tuning_data
|
37 |
-
β β β βββfine_tuning_data_detic.csv: Fine-tuning data prepared by the prompt engineering module using DETIC detector.
|
38 |
-
β β β βββfine_tuning_data_yolov5.csv: Fine-tuning data prepared by the prompt engineering module using YOLOv5. detector.
|
39 |
| βββ results
|
40 |
-
β β βββ Demo_Images: Contains a pool of images used for the demo app.
|
41 |
β β βββ evaluation.py: Provides a comprehensive framework for evaluating the KB-VQA model.
|
42 |
β β βββ demo.py: Provides a comprehensive framework for visualizing and demonstrating the results of the KB-VQA evaluation.
|
43 |
β β βββ evaluation_results.xlsx : This file contains all the evaluation results based on the evaluation data.
|
44 |
| βββ tabs
|
45 |
-
β β βββ home.py: Displays an introduction to the application with brief background along with the demo tools
|
46 |
β β βββ results.py: Manages the interactive Streamlit demo for visualizing model evaluation results and analysis.
|
47 |
-
β β βββ run_inference.py: Responsible for the 'run inference' tool to test and use the fine-tuned models.
|
48 |
β β βββ model_arch.py: Displays the model architecture and accompanying abstract and design details
|
49 |
β β βββ dataset_analysis.py: Provides tools for visualizing dataset analyses.
|
50 |
-
| βββ utilities
|
51 |
β β βββ ui_manager.py: Manages the user interface for the Streamlit application, handling the creation and navigation of various tabs.
|
52 |
β β βββ gen_utilities.py: Provides a collection of utility functions and classes commonly used across various parts
|
53 |
| βββ config (All Configurations files are kept separated and stored as ".py" for easy reading - this will change after the project submission.)
|
@@ -61,4 +62,5 @@ KB-VQA
|
|
61 |
βββ app.py: main entry point for streamlit - first page in the streamlit app)
|
62 |
βββ README.md (readme - this file)
|
63 |
βββ requirements.txt: Requirements file for the whole project that includes all the requirements for running the demo app on the HuggingFace space environment.
|
64 |
-
```
|
|
|
|
12 |
|
13 |
|
14 |
|
15 |
+
"""
|
16 |
|
17 |
## Project File Structure
|
18 |
|
19 |
```
|
20 |
KB-VQA
|
21 |
+
βββ Files: Various files required for the demo such as samples images, dissertation report ..etc.
|
22 |
βββ models
|
23 |
| βββ deformable-detr-detic: DETIC Object Detection Model.
|
24 |
| βββ yolov5: YOLOv5 Object Detection Model.baseline)
|
|
|
35 |
β β βββ fine_tuner.py: Main Fine-Tuning Script for LLaMa-2 Chat models.
|
36 |
β β βββ fine_tuning_data_handler.py: Handles and prepares the data for fine-tuning LLaMA-2 Chat models.
|
37 |
β β βββ fine_tuning_data
|
38 |
+
β β β βββfine_tuning_data_detic.csv: Fine-tuning data prepared by the prompt engineering module using DETIC detector.
|
39 |
+
β β β βββfine_tuning_data_yolov5.csv: Fine-tuning data prepared by the prompt engineering module using YOLOv5. detector.
|
40 |
| βββ results
|
41 |
+
β β βββ Demo_Images: Contains a pool of images used for the demo app.
|
42 |
β β βββ evaluation.py: Provides a comprehensive framework for evaluating the KB-VQA model.
|
43 |
β β βββ demo.py: Provides a comprehensive framework for visualizing and demonstrating the results of the KB-VQA evaluation.
|
44 |
β β βββ evaluation_results.xlsx : This file contains all the evaluation results based on the evaluation data.
|
45 |
| βββ tabs
|
46 |
+
β β βββ home.py: Displays an introduction to the application with brief background along with the demo tools description.
|
47 |
β β βββ results.py: Manages the interactive Streamlit demo for visualizing model evaluation results and analysis.
|
48 |
+
β β βββ run_inference.py: Responsible for the 'run inference' tool to test and use the fine-tuned models.
|
49 |
β β βββ model_arch.py: Displays the model architecture and accompanying abstract and design details
|
50 |
β β βββ dataset_analysis.py: Provides tools for visualizing dataset analyses.
|
51 |
+
| βββ utilities
|
52 |
β β βββ ui_manager.py: Manages the user interface for the Streamlit application, handling the creation and navigation of various tabs.
|
53 |
β β βββ gen_utilities.py: Provides a collection of utility functions and classes commonly used across various parts
|
54 |
| βββ config (All Configurations files are kept separated and stored as ".py" for easy reading - this will change after the project submission.)
|
|
|
62 |
βββ app.py: main entry point for streamlit - first page in the streamlit app)
|
63 |
βββ README.md (readme - this file)
|
64 |
βββ requirements.txt: Requirements file for the whole project that includes all the requirements for running the demo app on the HuggingFace space environment.
|
65 |
+
```
|
66 |
+
"""
|