티스토리 뷰

https://github.com/ultralytics/ultralytics/pull/2289

 

use clamp_ instead of clamp for an correct In-place operation by developer0hye · Pull Request #2289 · ultralytics/ultralytics

🤖 Generated by Copilot at d086fc0 Summary 📉⚡🎯 Improved memory efficiency of get_targets function in ultralytics/yolo/utils/tal.py by using in-place tensor operations. This function assigns labels ...

github.com

 

label assignment 과정에서 class index를 one hot encoding하기전에 clamp를 해주는데 기존 구현의 경우 in-place operation이 아니여서 쓸모없는 연산이 실행되고 있었다. 이걸 알맞게 수정해줬다.

 

코드 보면 label assignment 를 어떤 철학을 갖고 디자인했는지는 보이는데... 이걸 batch computation 으로 아주 잘 구현해놔갖고 오히려 코드를 100% 이해하기가 아직은 힘들다 좀 더 들여다봐야겠다.

댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/01   »
1 2 3 4
5 6 7 8 9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31
글 보관함