SeoJunn commited on
Commit
60a2fa3
1 Parent(s): e1c76b3

Refactor : 에러 나는 부분 예측코드

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from matplotlib import gridspec
4
  import matplotlib.pyplot as plt
5
  import numpy as np
6
  from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
7
  import torch
8
  import tensorflow as tf
9
  from PIL import ImageDraw
 
4
  import matplotlib.pyplot as plt
5
  import numpy as np
6
  from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
7
+ from transformers import DetrImageProcessor, DetrForObjectDetection
8
  import torch
9
  import tensorflow as tf
10
  from PIL import ImageDraw