logo
Browse Source

update training dataset infos in readme

main
ChengZi 2 years ago
parent
commit
5741e1fd67
  1. 6
      README.md

6
README.md

@ -121,15 +121,17 @@ If you want to fine tune this operator, make sure your timm version is 0.4.12, a
! python -m pip install tqdm augly timm==0.4.12 pytorch-metric-learning==0.9.99 ! python -m pip install tqdm augly timm==0.4.12 pytorch-metric-learning==0.9.99
``` ```
### Download dataset
Your need to download dataset from [Facebook AI Image Similarity Challenge: Descriptor Track](https://www.drivendata.org/competitions/79/competition-image-similarity-1-dev/), and use the `./training_images` folder as your training images root. It requires about 165G space.
### Get start to fine-tune
### Get started to fine-tune
Just call method op.train() and pass in your args. Just call method op.train() and pass in your args.
```python ```python
import towhee import towhee
op = towhee.ops.image_embedding.isc().get_op() op = towhee.ops.image_embedding.isc().get_op()
op.train(training_args={ op.train(training_args={
'train_data_dir': './your_training_images',
'train_data_dir': './training_images',
'distributed': False, 'distributed': False,
'output_dir': './output', 'output_dir': './output',
'gpu': 0, 'gpu': 0,

Loading…
Cancel
Save