ultralytics / ultralytics

Ultralytics YOLO11 🚀
https://docs.ultralytics.com
GNU Affero General Public License v3.0
31.62k stars 6.07k forks source link

about predict parameters and #15647

Open yeonhyochoi opened 2 months ago

yeonhyochoi commented 2 months ago

Search before asking

Question

Hello, thank you always for your hard work.

I'm applying the yolov8 detection model to my personal data set, and looking at the results.png picture, it looks like it's learning really well, but there are too many FPs. I am learning this way.

results

  1. The learning data set is 300,000 images with lesions, 30,000 images without lesions (10%), and the lesion is a single class!

  2. Training from scratch with pretrained =False

  3. 300epoch training

And the actual test is measuring TP and FP accuracy for each frame in the video.

This is my actual test code. Here's what I'm curious about: image image

  1. Is there a need to change to RGB using cv2.cvtColor?
  2. Is it necessary to apply normalization to changed frames?

I've seen this question a lot in your documents, but I'm asking it again. It seems that your documents automatically pre-process the frames even if they are loaded that way. The reason I ask this question is because I am curious about whether there is a difference between the data being trained and the data being tested during the preprocessing process.

This problem hasn't been resolved for a month. I would really appreciate your help!

Additional

No response

Y-T-G commented 2 months ago

To avoid default preprocessing, you can pass the preprocessed frames as tensors.