Merge branch 'master' into dev
commit
4a3f8d02c5
|
@ -15,9 +15,6 @@ def main():
|
|||
|
||||
# build the model from a config file and a checkpoint file
|
||||
model = init_model(args.config, args.checkpoint, device=args.device)
|
||||
for i, stage in enumerate(model.backbone.stages):
|
||||
print(i, len(stage))
|
||||
print(i, stage[0])
|
||||
# test a single image
|
||||
result = inference_model(model, args.img)
|
||||
# show the results
|
||||
|
|
Loading…
Reference in New Issue