Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

coco dataset support, automatic aws weight upload #54

Merged
merged 16 commits into from
Nov 9, 2021
Merged

coco dataset support, automatic aws weight upload #54

merged 16 commits into from
Nov 9, 2021

Conversation

fcakyon
Copy link
Owner

@fcakyon fcakyon commented Oct 27, 2021 •
edited
Loading

  • Start a training using a COCO formatted dataset:
# data.yml
train_json_path: "train.json"
train_image_dir: "train_image_dir/"
val_json_path: "val.json"
val_image_dir: "val_image_dir/"
$ yolov5 train --data data.yaml --weights yolov5s.pt
  • Automatically upload weights to AWS S3 (with Neptune.AI artifact tracking integration):
export AWS_ACCESS_KEY_ID=YOUR_KEY
export AWS_SECRET_ACCESS_KEY=YOUR_KEY
$ yolov5 train --data data.yaml --weights yolov5s.pt --s3_dir YOUR_S3_FOLDER_DIRECTORY

@fcakyon fcakyon added the enhancement New feature or request label Oct 27, 2021
@fcakyon fcakyon self-assigned this Oct 27, 2021
@fcakyon fcakyon changed the title add automatic aws weight upload coco dataset support, automatic aws weight upload Nov 2, 2021
@fcakyon fcakyon merged commit 07c9cab into main Nov 9, 2021
ameicler pushed a commit to ameicler/yolov5-pip that referenced this pull request Apr 18, 2022
@fcakyon
coco dataset support, automatic aws weight upload (fcakyon#54)
3ccad36 
* ignore neptune logs

* ignore neptune logs

* add s3 weight upload to readme

* add s3 file upload util

* integrate automatic s3 upload

* add missing boto3 requirement

* update logging

* better exception handling

* update s3 weight dir

* add coco styled data support for training

* add sahi dependency

* update sahi version

* fix wandb logging error

* fix category based ap logging

* minor fix
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@fcakyon fcakyon

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@fcakyon