|
@@ -150,7 +150,7 @@ class Trainer(BaseTrainer):
|
|
|
def writer_predict_result(self, img, result, epoch):
|
|
def writer_predict_result(self, img, result, epoch):
|
|
|
img = img.cpu().detach()
|
|
img = img.cpu().detach()
|
|
|
im = img.permute(1, 2, 0) # [512, 512, 3]
|
|
im = img.permute(1, 2, 0) # [512, 512, 3]
|
|
|
- self.writer.add_image("ori", im, epoch, dataformats="HWC")
|
|
|
|
|
|
|
+ self.writer.add_image("z-ori", im, epoch, dataformats="HWC")
|
|
|
|
|
|
|
|
boxed_image = draw_bounding_boxes((img * 255).to(torch.uint8), result["boxes"],
|
|
boxed_image = draw_bounding_boxes((img * 255).to(torch.uint8), result["boxes"],
|
|
|
colors="yellow", width=1)
|
|
colors="yellow", width=1)
|
