CHE-Master / README.md
Elfenreigen's picture
Update README.md
b7a3673 verified
# README
## 0. Configuration File
The config file is `./configs/example.yaml`. You can modify `test_file` or `finetune`, which correspond to the CSV file location and the pre-trained model checkpoint location, respectively.
## 1. Model Weight Files and Dataset Configuration
### Model Weight Files
The model weight files are located in the `./configs` folder. Ensure that these files are correctly loaded when running the model.
### CSV File
The CSV file contains the paths to the test images and their corresponding labels. This file is used for testing the model.
### Modifying `test_example.py`
#### Adjusting `Chestxray14_Dataset` Class
In the `Chestxray14_Dataset` class within `test_example.py`, you need to adjust the column indices according to the structure of your CSV file. Specifically, ensure that the columns for image paths and labels are correctly referenced.
#### Updating `valid_on` Method
The `valid_on` method in `test_example.py` contains a `text_list` that should correspond to the labels in your CSV file. Make sure that the `text_list` is updated to match the labels in your dataset.
## 2. Model Output
### Prediction Results
The model will generate prediction results for each image and each class. These results will be saved as `.npy` files.
### Output Location
The `.npy` files will be stored in subfolders within the `./results` directory. Each subfolder corresponds to a specific run or configuration of the model.