pyesonekyaw commited on
Commit
a3e2ae9
1 Parent(s): 80288b5
Files changed (1) hide show
  1. models/common.py +1 -3
models/common.py CHANGED
@@ -15,7 +15,6 @@ import pandas as pd
15
  import requests
16
  import torch
17
  import torch.nn as nn
18
- from IPython.display import display
19
  from PIL import Image
20
  from torch.cuda import amp
21
 
@@ -529,8 +528,7 @@ class Detections:
529
  s += '(no detections)'
530
 
531
  im = Image.fromarray(im.astype(np.uint8)) if isinstance(im, np.ndarray) else im # from np
532
- if show:
533
- display(im) if is_notebook() else im.show(self.files[i])
534
  if save:
535
  f = self.files[i]
536
  im.save(save_dir / f) # save
 
15
  import requests
16
  import torch
17
  import torch.nn as nn
 
18
  from PIL import Image
19
  from torch.cuda import amp
20
 
 
528
  s += '(no detections)'
529
 
530
  im = Image.fromarray(im.astype(np.uint8)) if isinstance(im, np.ndarray) else im # from np
531
+
 
532
  if save:
533
  f = self.files[i]
534
  im.save(save_dir / f) # save