Skip to content

How to set the dataset to BTech? #161

Answered by samet-akcay
brm738 asked this question in Q&A
Discussion options

You must be logged in to vote

Hi @brm738, the data section in the config.yaml file would be as follows:

dataset:
  name: btech #options: [mvtec, btech, folder]
  format: btech
  path: ./datasets/BTech
  category: '01'
  task: segmentation
  image_size: 256
  train_batch_size: 32
  test_batch_size: 32
  num_workers: 8
  transform_config: null
  create_validation_set: false
  tiling:
    apply: false
    tile_size: null
    stride: null
    remove_border_count: 0
    use_random_tiling: False
    random_tile_count: 16

For instance, if you set the above configurations in anomalib/models/patchcore/config.yaml, and run python tools/train.py --model patchcore, you could train BTech dataset.

I've noticed that it throws an err…

Replies: 3 comments 3 replies

Comment options

You must be logged in to vote
1 reply
@brm738
Comment options

Answer selected by brm738
Comment options

You must be logged in to vote
1 reply
@samet-akcay
Comment options

Comment options

You must be logged in to vote
1 reply
@brm738
Comment options

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants