*config setting (cascade rcnn 의 경우)
- pretrained = backbone인 ResNeXt의 웨이트 경로 입력
- load_from = transfer learning 할 경우, 이전에 학습시켰던 웨이트의 경로 입력
- num_classes = transfer learning 할 때 원하는 클래스 수 입력
반응형
'스타트업 > AI' 카테고리의 다른 글
| [AI] Hue(Hadoop User Experience) (0) | 2020.07.30 |
|---|---|
| [AI] pytorch shared memory error (0) | 2020.07.22 |
| [AI] 리눅스 명령어 (0) | 2020.07.21 |
| [AI] CRF 알고리즘 (Conditional Random Field) (0) | 2020.07.16 |
| [AI] Multi Head Attention (0) | 2020.07.16 |