Skip to content

Commit b225ba1

Browse files
Merge pull request #5 from renan-siqueira/documentation/DocFiles
Documentation
2 parents 2880ca9 + 7936720 commit b225ba1

File tree

2 files changed

+68
-1
lines changed

2 files changed

+68
-1
lines changed

LICENSE

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
MIT License
2+
3+
Copyright (c) 2023 renan-siqueira
4+
5+
Permission is hereby granted, free of charge, to any person obtaining a copy
6+
of this software and associated documentation files (the "Software"), to deal
7+
in the Software without restriction, including without limitation the rights
8+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9+
copies of the Software, and to permit persons to whom the Software is
10+
furnished to do so, subject to the following conditions:
11+
12+
The above copyright notice and this permission notice shall be included in all
13+
copies or substantial portions of the Software.
14+
15+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21+
SOFTWARE.

README.md

Lines changed: 47 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,47 @@
1-
# Autoencoder Project
1+
# Autoencoder Project
2+
3+
A simple implementation of an autoencoder using PyTorch.
4+
5+
This project aims to provide a foundational structure to understand, train, and evaluate autoencoders on 64x64 images.
6+
7+
## Features
8+
9+
- Train an autoencoder on your dataset of images.
10+
- Visualize the reconstructions of the autoencoder.
11+
- Evaluate the model on a separate validation set.
12+
- Save and load model functionality.
13+
14+
## Getting Started
15+
16+
### Prerequisites
17+
18+
- Python 3.x
19+
- PyTorch
20+
- torchvision
21+
- PIL
22+
- matplotlib
23+
24+
### Installation
25+
26+
1. Clone the repository:
27+
28+
```bash
29+
git clone https://github.com/renan-siqueira/autoencoder-project.git
30+
```
31+
2. Navigate to the project directory and install the required libraries:
32+
33+
```bash
34+
cd autoencoder-project
35+
pip install -r requirements.txt
36+
```
37+
38+
## Usage
39+
40+
1. Modify settings/settings.py to point to your training and validation dataset.
41+
2. To train the autoencoder, simply run:
42+
43+
```bash
44+
python run.py
45+
```
46+
47+
By default, this will train a new model. If you wish to use a pre-trained model, modify the main method in run.py.

0 commit comments

Comments
 (0)