Skip to content

Commit

Permalink
Merge pull request #26 from caetas/dev
Browse files Browse the repository at this point in the history
ImageNet support and wandb can be disabled
  • Loading branch information
caetas authored Aug 9, 2024
2 parents 0fea9d3 + c773323 commit 83fc7a1
Show file tree
Hide file tree
Showing 81 changed files with 1,432,031 additions and 581 deletions.
17 changes: 17 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -161,6 +161,15 @@ The following datasets are ready to be used to train and sample from the provide
- DTD [`Source`](https://arxiv.org/abs/1311.3618)
- TinyImageNet [`Source`](https://cs231n.stanford.edu/reports/2015/pdfs/yle_project.pdf) **MANUAL DOWNLOAD REQUIRED** [`Link`](https://www.kaggle.com/datasets/nikhilshingadiya/tinyimagenet200)
- Horse2Zebra [`Source`](https://arxiv.org/abs/1703.10593v6) **MANUAL DOWNLOAD REQUIRED** [`Link`](https://www.kaggle.com/datasets/balraj98/horse2zebra-dataset)
- ImageNet-1k [`Source`](https://ieeexplore.ieee.org/abstract/document/5206848) **MANUAL DOWNLOAD REQUIRED** [`Link`](https://www.kaggle.com/c/imagenet-object-localization-challenge/data) **AND PRE-PROCESSING FOLLOWING THE INSTRUCTIONS BELOW**

#### ImageNet-1K Pre-Processing

ImageNet-1K must be downloaded from [Kaggle](https://www.kaggle.com/c/imagenet-object-localization-challenge/data). After downloading it:
1. Unzip the Folder.
2. Rename the folder `ILSVRC/Data/CLS-LOC` as `imagenet`.
3. Move `imagenet` only to [`data/raw`](data/raw). The other folders can be deleted.
4. Run the [`process_imagenet.py`](src/generativezoo/process_imagenet.py) script. This operation takes a while due to de dataset size.

## Tracking

Expand All @@ -178,6 +187,14 @@ The code examples are setup to use [Weights & Biases](https://wandb.ai/home) as
wandb login
```

**If you want to fully disable Weights & Biases during training, use the flag `--no_wandb`.**

**If you want to turn off syncing with the server but want to retain the local copy, run:**

```bash
wandb offline
```

## Repository Information

### Documentation
Expand Down
2 changes: 1 addition & 1 deletion VERSION
Original file line number Diff line number Diff line change
@@ -1 +1 @@
0.20.0
1.1.0
Loading

0 comments on commit 83fc7a1

Please sign in to comment.