Skip to content

update dataset format and fix some bug #90

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.

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

Open
wants to merge 13 commits into
base: main
Choose a base branch
from

Conversation

yjmm10
Copy link

@yjmm10 yjmm10 commented Aug 28, 2024

  1. ModelValidator align ModelTrainer, ModelValidator
  2. add log of optimizer
  3. get image_id map during validation
  4. update the structure of folder, support the image_root and label_root in different phase
  5. get_metadata will get the image_file not the image_id
  6. add the mode include "detection", "segmentation", default "detection"
  7. fix the good_epoch

yongjian_zhang added 13 commits August 14, 2024 03:44
2. add log of optimizer
3. get image_id map during validation
4. update the structure of folder, support the image_root and label_root in different phase
5. get_metadata will get the image_file not the image_id
6. add the mode include "detection", "segmentation", default "detection"
7. fix the good_epoch
merge coco object, duplicate data is not merged
…e the more ap with predict.json

fix: GradScaler should be used under gpu
update: use predcit.json to compute the AP logic
2. optimize the program for calculating AP and support more AP index
3. fix the choice of mode
add: more test_calculate_map case
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant