-
Notifications
You must be signed in to change notification settings - Fork 35
Open
Description
Hi, I have convert my coco annotations using coco_tfrecord_creator.py.
Unfortunatelly train.py throws me this:
tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimension -66 must be >= 0
[[{{node parser/zeros_3}}]]
Arguments I used:
- convert to TFRecord
python -m data.coco_tfrecord_creator -train_image_dir Dataset/train/images -val_image_dir Dataset/val/images -train_annotations_file Dataset/train/annotations/instances_default.json -val_annotations_file Dataset/val/annotations/instances_default.json -output_dir ./data/coco
-training:
python train.py -name 'coco' -tfrecord_dir './data' -weights './weights' -batch_size '4' -momentum '0.9' -weight_decay '0.0005' -print_interval '10' -save_interval '10'
You could also add text file with requirements :) . I'm using tensorflow-gpu 2.3.0 version and custom dataset.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels