https://github.com/tyleryep/lung-xray
Stanford CS 271 Project
https://github.com/tyleryep/lung-xray
Last synced: 11 months ago
JSON representation
Stanford CS 271 Project
- Host: GitHub
- URL: https://github.com/tyleryep/lung-xray
- Owner: TylerYep
- Created: 2020-02-18T04:21:38.000Z (over 6 years ago)
- Default Branch: main
- Last Pushed: 2023-07-26T06:01:03.000Z (almost 3 years ago)
- Last Synced: 2025-06-10T11:44:30.594Z (about 1 year ago)
- Language: Jupyter Notebook
- Size: 1.97 MB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# lung-xray
Stanford CS 271 Project
## TODO:
1. Change data to PyTorch Dataset
- [DONE] Fill in dataset.py MyDataset
2. Visualize/Explore data
- viz.py
3. Add model
- models/pnunet.py
4. Change loss/optimization
- train.py
5. Train/Tune - Try different models.
## How to Run
1. Start tensorboard using the `checkpoints/` folder with `tensorboard --logdir=checkpoints/`
2. Start and stop training using `python train.py --checkpoint=`. The code should automatically resume training at the previous epoch and continue logging to the previous tensorboard.
3. Run `python test.py --checkpoint=` to get final predictions.
# Directory Structure
- checkpoints/ (Only created once you run train.py)
- data/
- metrics/
- models/
- layers/
- ...
- visualizers/
- args.py (Modify default hyperparameters here)
- dataset.py (Create Dataset here)
- metric_tracker.py
- models.py (You may opt to keep all your models in one place instead)
- preprocess.py (Do any preprocessing steps you want before loading the data)
- test.py
- train.py
- util.py
- viz.py (Create more visualizations if necessary)