Skip to content

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