Update README.md
Browse files
README.md
CHANGED
|
@@ -49,9 +49,9 @@ Code
|
|
| 49 |
### 2. Viewing the Results (Using the Pre-trained Model)
|
| 50 |
|
| 51 |
1. Run the V2 viewer script:
|
| 52 |
-
|
| 53 |
python pkas_cal_viewer_gemini2.py
|
| 54 |
-
|
| 55 |
2. In the GUI:
|
| 56 |
- Select the COCO image and annotation paths you downloaded.
|
| 57 |
- Click **"Load V2 Model"** and select the `calcium_bridge_eeg_model_v2.pth` file you downloaded from Hugging Face.
|
|
@@ -60,11 +60,13 @@ Code
|
|
| 60 |
### 3. Training Your Own Model (Optional)
|
| 61 |
|
| 62 |
1. Run the V2 training script:
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
|
|
|
|
|
|
| 68 |
|
| 69 |
## A Note on Interpretation
|
| 70 |
|
|
|
|
| 49 |
### 2. Viewing the Results (Using the Pre-trained Model)
|
| 50 |
|
| 51 |
1. Run the V2 viewer script:
|
| 52 |
+
|
| 53 |
python pkas_cal_viewer_gemini2.py
|
| 54 |
+
|
| 55 |
2. In the GUI:
|
| 56 |
- Select the COCO image and annotation paths you downloaded.
|
| 57 |
- Click **"Load V2 Model"** and select the `calcium_bridge_eeg_model_v2.pth` file you downloaded from Hugging Face.
|
|
|
|
| 60 |
### 3. Training Your Own Model (Optional)
|
| 61 |
|
| 62 |
1. Run the V2 training script:
|
| 63 |
+
|
| 64 |
+
|
| 65 |
+
python pkas_cal_trainer_gemini.py
|
| 66 |
+
|
| 67 |
+
3. In the GUI, select your COCO image and annotation paths.
|
| 68 |
+
4. Click **"Train Extended Model (V2)"**.
|
| 69 |
+
5. A new file named `calcium_bridge_eeg_model_v2.pth` will be saved with the best-performing model from your training run. You can then load this file into the viewer.
|
| 70 |
|
| 71 |
## A Note on Interpretation
|
| 72 |
|