BMS, YoloE 관련 기능 추가 - Githubissues

ML-TANGO / TANGO

public repo for TANGO (Target Aware No-code neural network Generation and Operation framework)

Other

68 stars 20 forks source link

BMS, YoloE 관련 기능 추가 #100

Closed iksooman closed 11 months ago

iksooman commented 11 months ago

PR 타입

하나 이상의 PR 타입을 선택해주세요.

[x] 기능 추가
[ ] 기능 삭제
[ ] 버그 수정
[x] 의존성, 환경 변수, 빌드 관련 코드 업데이트
[ ] 기타

PR전 체크 사항

PR전 아래 사항을 확인하였는지 체크하시고 PR 생성 바랍니다.

[x] 자신의 브랜치로 main 브랜치 최신 내용 반영후 동작 테스트 완료
[x] 다른 작업자의 폴더내 파일 수정하지 않음. --> BMS, YoloE 관련 코드 수정되었습니다.
[ ] docker-compose.yml 수정한 경우, 수정에 따른 영향 평가 --> docker-compose 수정하지 않았습니다.

관련 이슈

PR 반영 브랜치

ETRI_sis_BMS -> main

변경 사항

BMS가 중앙대의 supernet 모델을 사용해서 batch size를 계산할 수 있도록 기능을 추가하였습니다.
YoloE가 onnx 파일도 생성할 수 있도록 기능을 추가하였습니다.

테스트 결과