nehulagrawal
commited on
Commit
•
eebde71
1
Parent(s):
c461bfe
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,184 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
tags:
|
3 |
+
- ultralyticsplus
|
4 |
+
- yolov8
|
5 |
+
- ultralytics
|
6 |
+
- yolo
|
7 |
+
- vision
|
8 |
+
- object-detection
|
9 |
+
- pytorch
|
10 |
+
- finance
|
11 |
+
- stock market
|
12 |
+
- candlesticks
|
13 |
+
- pattern recognition
|
14 |
+
- option trading
|
15 |
+
- chart reader
|
16 |
+
- future stock prediction
|
17 |
+
- trends prediction
|
18 |
+
library_name: ultralytics
|
19 |
+
library_version: 8.0.43
|
20 |
+
inference: false
|
21 |
+
model-index:
|
22 |
+
- name: foduucom/stockmarket-future-prediction
|
23 |
+
results:
|
24 |
+
- task:
|
25 |
+
type: object-detection
|
26 |
+
metrics:
|
27 |
+
- type: precision
|
28 |
+
value: 0.663
|
29 |
+
name: [email protected](box)
|
30 |
+
language:
|
31 |
+
- en
|
32 |
+
pipeline_tag: object-detection
|
33 |
+
---
|
34 |
+
|
35 |
+
|
36 |
+
# Model Card for YOLOv8s Stock Market future trends prediction on Live Trading Video Data
|
37 |
+
|
38 |
+
## Model Summary
|
39 |
+
|
40 |
+
The YOLOv8s Stock Market future trends prediction model is an object detection model based on the YOLO (You Only Look Once) framework. It is designed to detect various chart patterns in real-time stock market trading video data. The model aids traders and investors by automating the analysis of chart patterns, providing timely insights for informed decision-making. The model has been fine-tuned on a diverse dataset and achieved high accuracy in detecting and classifying stock market future trend detection in live trading scenarios.
|
41 |
+
|
42 |
+
## Model Details
|
43 |
+
|
44 |
+
### Model Description
|
45 |
+
The YOLOv8s Stock Market future trends prediction model offers a transformative solution for traders and investors by enabling real-time detection of crucial chart patterns within live trading video data. As stock markets evolve rapidly, this model's capabilities empower users with timely insights, allowing them to make informed decisions with speed and accuracy.
|
46 |
+
|
47 |
+
The model seamlessly integrates into live trading systems, providing instant trends prediction and classification. By leveraging advanced bounding box techniques and pattern-specific feature extraction, the model excels in identifying patterns such as 'Down','Up'. This enables traders to optimize their strategies, automate trading decisions, and respond to market trends in real-time.
|
48 |
+
|
49 |
+
To facilitate integration into live trading systems or to inquire about customization, please contact us at [email protected]. Your collaboration and feedback are instrumental in refining and enhancing the model's performance in dynamic trading environments.
|
50 |
+
|
51 |
+
- **Developed by:** FODUU AI
|
52 |
+
- **Model type:** Object Detection
|
53 |
+
- **Task:** Stock Market future trends prediction on Live Trading Video Data
|
54 |
+
|
55 |
+
The YOLOv8s Stock Market Pattern Detection model is designed to adapt to the fast-paced nature of live trading environments. Its ability to operate on real-time video data allows traders and investors to harness pattern-based insights without delay.
|
56 |
+
|
57 |
+
### Supported Labels
|
58 |
+
|
59 |
+
```
|
60 |
+
['Down','Up']
|
61 |
+
```
|
62 |
+
|
63 |
+
## Uses
|
64 |
+
|
65 |
+
### Direct Use
|
66 |
+
|
67 |
+
The YOLOv8s Stock Market future trends prediction model can be directly integrated into live trading systems to provide real-time detection and classification of chart patterns or classify the upcoming trends. Traders can utilize the model's insights for timely decision-making.
|
68 |
+
|
69 |
+
### Downstream Use
|
70 |
+
|
71 |
+
The model's real-time capabilities can be leveraged to automate trading strategies, generate alerts for specific patterns or trends, and enhance overall trading performance.
|
72 |
+
|
73 |
+
### Out-of-Scope Use
|
74 |
+
|
75 |
+
The model is not designed for unrelated object detection tasks or scenarios outside the scope of stock market trends prediction in live trading video data.
|
76 |
+
|
77 |
+
## Bias, Risks, and Limitations
|
78 |
+
|
79 |
+
The YOLOv8s Stock Market future prediction model may exhibit some limitations and biases:
|
80 |
+
|
81 |
+
- Performance may be affected by variations in video quality, lighting conditions, and pattern complexity within live trading data.
|
82 |
+
- Rapid market fluctuations and noise in video data may impact the model's accuracy and responsiveness.
|
83 |
+
- Market-specific patterns or anomalies not well-represented in the training data may pose challenges for detection.
|
84 |
+
|
85 |
+
### Recommendations
|
86 |
+
|
87 |
+
Users should be aware of the model's limitations and potential biases. Thorough testing and validation within live trading simulations are advised before deploying the model in real trading environments.
|
88 |
+
|
89 |
+
## How to Get Started with the Model
|
90 |
+
|
91 |
+
To begin using the YOLOv8s Stock Market future prediction model on live trading video data, follow these steps:
|
92 |
+
```bash
|
93 |
+
pip install ultralyticsplus==0.0.28 ultralytics==8.0.43
|
94 |
+
```
|
95 |
+
|
96 |
+
- Load model and perform real-time prediction:
|
97 |
+
|
98 |
+
```python
|
99 |
+
from ultralyticsplus import YOLO, render_result
|
100 |
+
import cv2
|
101 |
+
|
102 |
+
# load model
|
103 |
+
model = YOLO('foduucom/stockmarket-future-prediction')
|
104 |
+
|
105 |
+
# set model parameters
|
106 |
+
model.overrides['conf'] = 0.25 # NMS confidence threshold
|
107 |
+
model.overrides['iou'] = 0.45 # NMS IoU threshold
|
108 |
+
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
109 |
+
model.overrides['max_det'] = 1000 # maximum number of detections per image
|
110 |
+
|
111 |
+
# initialize video capture
|
112 |
+
# Open the video file
|
113 |
+
video_path = "path/to/your/video/file.mp4"
|
114 |
+
cap = cv2.VideoCapture(video_path)
|
115 |
+
|
116 |
+
# Loop through the video frames
|
117 |
+
while cap.isOpened():
|
118 |
+
# Read a frame from the video
|
119 |
+
success, frame = cap.read()
|
120 |
+
|
121 |
+
if success:
|
122 |
+
# Run YOLOv8 inference on the frame
|
123 |
+
results = model(frame)
|
124 |
+
|
125 |
+
# Visualize the results on the frame
|
126 |
+
annotated_frame = results[0].plot()
|
127 |
+
|
128 |
+
# Display the annotated frame
|
129 |
+
cv2.imshow("YOLOv8 Inference", annotated_frame)
|
130 |
+
|
131 |
+
# Break the loop if 'q' is pressed
|
132 |
+
if cv2.waitKey(1) & 0xFF == ord("q"):
|
133 |
+
break
|
134 |
+
else:
|
135 |
+
# Break the loop if the end of the video is reached
|
136 |
+
break
|
137 |
+
|
138 |
+
# Release the video capture object and close the display window
|
139 |
+
cap.release()
|
140 |
+
cv2.destroyAllWindows()
|
141 |
+
```
|
142 |
+
|
143 |
+
## Training Details
|
144 |
+
|
145 |
+
### Training Data
|
146 |
+
|
147 |
+
The model is trained on a diverse dataset containing stock market chart images with various chart patterns, capturing different market conditions and scenarios.
|
148 |
+
|
149 |
+
### Training Procedure
|
150 |
+
|
151 |
+
The training process involves extensive computation and is conducted over multiple epochs. The model's weights are adjusted to minimize detection loss and optimize performance for stock market pattern detection.
|
152 |
+
|
153 |
+
#### Metrics
|
154 |
+
|
155 |
+
- [email protected] (box):
|
156 |
+
- All patterns: 0.90
|
157 |
+
- Individual patterns: Varies based on pattern type
|
158 |
+
|
159 |
+
### Model Architecture and Objective
|
160 |
+
|
161 |
+
The YOLOv8s architecture incorporates modifications tailored to stock market future prediction. It features a specialized backbone network, self-attention mechanisms, and trends-specific feature extraction modules.
|
162 |
+
|
163 |
+
### Compute Infrastructure
|
164 |
+
|
165 |
+
#### Hardware
|
166 |
+
|
167 |
+
NVIDIA GeForce RTX 3080 card
|
168 |
+
|
169 |
+
#### Software
|
170 |
+
|
171 |
+
The model was trained and fine-tuned using a Jupyter Notebook environment.
|
172 |
+
|
173 |
+
## Model Card Contact
|
174 |
+
|
175 |
+
For inquiries and contributions, please contact us at [email protected].
|
176 |
+
|
177 |
+
```bibtex
|
178 |
+
@ModelCard{
|
179 |
+
author = {Nehul Agrawal and
|
180 |
+
Pranjal Singh Thakur},
|
181 |
+
title = {YOLOv8s Stock Market future prediction on Live Trading Video Data},
|
182 |
+
year = {2023}
|
183 |
+
}
|
184 |
+
```
|