Update get_dr_txt.py

get_dr_txt.py

@@ -9,7 +9,7 @@ from keras.layers import Input
 from keras.applications.imagenet_utils import preprocess_input
 from keras import backend as K
 from utils.utils import letterbox_image
-from nets.yolo3 import yolo_body,yolo_eval
+from nets.yolo4 import yolo_body,yolo_eval
 import colorsys
 import numpy as np
 import os
@@ -110,4 +110,4 @@ for image_id in image_ids:
     yolo.detect_image(image_id,image)
     print(image_id," done!")
     
-print("Conversion completed!")
\ No newline at end of file
+print("Conversion completed!")